使用yolov3_mobilenet_v1_fruit.yml训练出模型,在红米7上运行很慢
Created by: sdjkxhb2016
我把我自己的数据集放到PaddleDetection/dataset/fruit目录,yolov3_mobilenet_v1_fruit.yml 也没有修改,直接用下面的语句训练
!python -u tools/train.py -c configs/yolov3_mobilenet_v1_fruit.yml
--use_tb=True
--tb_log_dir=tb_fruit_dir/scalar
--eval
训练出的模型放到红米7上运行,可以正确检测,就是非常慢,我用Paddle_Lite 的SSD模型测试只要300ms,而用我的模型,要1300-1700ms ,我没有使用slim 压缩模型