背景:yolov5部署到android端之后,发现一个模型实现不了全部的功能,初期训练的和后面的标签不一致不过都是yolov5的,感觉可以合并下。这里主要讲述实现加载2个模型,在调用时候动态选择哪个模型
一:Init修改
1:重新定义一个网络
static ncnn::Net yolov51;
2:加载网络参数
yolov51.opt = opt;
3:注册
yolov51.register_custom_layer("YoloV5Focus", YoloV5Focus_layer_creator);
4:加载模型
yolov51.load_param(mgr, "yolov5.param");yolov51.load_model(mgr, "yolov5.bin");
二:Detect修改(根据输入参数选择使用哪个模型)
1:添加参数
jstring whdetect
2:参数类型转换
const char* chardata = env->GetStringUTFChars(whdetect, 0); // char* 转 string std::stri