使用v2.3 opt转换原始口罩模型生成.nb模型后检测不到人脸
Created by: yuhaoluo
opt是用v2.3的代码自行编译,原始模型是下载的pyramidbox_lite_mobile_mask。
由于我想做模型量化,所以需要自己转换模型。但是发现新版本转换生成.nb模型检测不到人脸。
我尝试先不量化模型,直接由原始模型使用opt转换,但还是检测不到人脸。代码直接使用原始的mask_detection.cc代码,只是将载入模型改成config.set_model_from_file(XXX)
opt转换命令如下:
./opt \
--model_file=pyramidbox_lite_mobile_mask/pyramidbox_lite/__model__ \
--param_file=pyramidbox_lite_mobile_mask/pyramidbox_lite/__param__ \
--optimize_out_type=naive_buffer \
--optimize_out=face_detect_arm \
--valid_targets=arm
./opt \
--model_file=pyramidbox_lite_mobile_mask/mask_detector/__model__ \
--param_file=pyramidbox_lite_mobile_mask/mask_detector/__param__ \
--optimize_out_type=naive_buffer \
--optimize_out=mask_cls_arm \
--valid_targets=arm