You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
95 lines
3.2 KiB
C++
95 lines
3.2 KiB
C++
8 years ago
|
#include <jni.h>
|
||
|
|
#include <string>
|
||
|
|
#include "include/Pipeline.h"
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
std::string jstring2str(JNIEnv* env, jstring jstr)
|
||
|
|
{
|
||
|
|
char* rtn = NULL;
|
||
|
|
jclass clsstring = env->FindClass("java/lang/String");
|
||
|
|
jstring strencode = env->NewStringUTF("GB2312");
|
||
|
|
jmethodID mid = env->GetMethodID(clsstring, "getBytes", "(Ljava/lang/String;)[B");
|
||
|
|
jbyteArray barr= (jbyteArray)env->CallObjectMethod(jstr,mid,strencode);
|
||
|
|
jsize alen = env->GetArrayLength(barr);
|
||
|
|
jbyte* ba = env->GetByteArrayElements(barr,JNI_FALSE);
|
||
|
|
if(alen > 0)
|
||
|
|
{
|
||
|
|
rtn = (char*)malloc(alen+1);
|
||
|
|
memcpy(rtn,ba,alen);
|
||
|
|
rtn[alen]=0;
|
||
|
|
}
|
||
|
|
env->ReleaseByteArrayElements(barr,ba,0);
|
||
|
|
std::string stemp(rtn);
|
||
|
|
free(rtn);
|
||
|
|
return stemp;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
extern "C" {
|
||
|
|
JNIEXPORT jlong JNICALL
|
||
|
|
Java_pr_platerecognization_PlateRecognition_InitPlateRecognizer(
|
||
|
|
JNIEnv *env, jobject obj,
|
||
|
|
jstring detector_filename,
|
||
|
|
jstring finemapping_prototxt, jstring finemapping_caffemodel,
|
||
|
|
jstring segmentation_prototxt, jstring segmentation_caffemodel,
|
||
|
|
jstring charRecognization_proto, jstring charRecognization_caffemodel) {
|
||
|
|
|
||
|
|
std::string detector_path = jstring2str(env, detector_filename);
|
||
|
|
std::string finemapping_prototxt_path = jstring2str(env, finemapping_prototxt);
|
||
|
|
std::string finemapping_caffemodel_path = jstring2str(env, finemapping_caffemodel);
|
||
|
|
std::string segmentation_prototxt_path = jstring2str(env, segmentation_prototxt);
|
||
|
|
std::string segmentation_caffemodel_path = jstring2str(env, segmentation_caffemodel);
|
||
|
|
std::string charRecognization_proto_path = jstring2str(env, charRecognization_proto);
|
||
|
|
std::string charRecognization_caffemodel_path = jstring2str(env, charRecognization_caffemodel);
|
||
|
|
|
||
|
|
|
||
|
|
pr::PipelinePR *PR = new pr::PipelinePR(detector_path,
|
||
|
|
finemapping_prototxt_path, finemapping_caffemodel_path,
|
||
|
|
segmentation_prototxt_path,
|
||
|
|
segmentation_caffemodel_path,
|
||
|
|
charRecognization_proto_path,
|
||
|
|
charRecognization_caffemodel_path);
|
||
|
|
|
||
|
|
return (jlong) PR;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
JNIEXPORT jstring JNICALL
|
||
|
|
Java_pr_platerecognization_PlateRecognition_SimpleRecognization(
|
||
|
|
JNIEnv *env, jobject obj,
|
||
|
|
jlong matPtr, jlong object_pr) {
|
||
|
|
pr::PipelinePR *PR = (pr::PipelinePR *) object_pr;
|
||
|
|
cv::Mat &mRgb = *(cv::Mat *) matPtr;
|
||
|
|
cv::Mat rgb;
|
||
|
|
cv::cvtColor(mRgb,rgb,cv::COLOR_RGBA2BGR);
|
||
|
|
cv::imwrite("/sdcard/demo.jpg",rgb);
|
||
|
|
|
||
|
|
std::vector<pr::PlateInfo> list_res= PR->RunPiplineAsImage(rgb);
|
||
|
|
std::string concat_results;
|
||
|
|
for(auto one:list_res)
|
||
|
|
{
|
||
|
|
if (one.confidence>0.7)
|
||
|
|
concat_results+=one.getPlateName()+",";
|
||
|
|
}
|
||
|
|
concat_results = concat_results.substr(0,concat_results.size()-1);
|
||
|
|
|
||
|
|
return env->NewStringUTF(concat_results.c_str());
|
||
|
|
|
||
|
|
}
|
||
|
|
JNIEXPORT jstring JNICALL
|
||
|
|
Java_pr_platerecognization_PlateRecognition_ReleasePlateRecognizer(
|
||
|
|
JNIEnv *env, jobject obj,
|
||
|
|
jlong object_re) {
|
||
|
|
// std::string hello = "Hello from C++";
|
||
|
|
pr::PipelinePR *PR = (pr::PipelinePR *) object_re;
|
||
|
|
delete PR;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|