1. 数据准备
2.安装cython_bbox失败
这里参考这位博主
博客地址
总结:手动下载,更改setup.py 配置 然后在该目录下重新用pip 安装
cython-bbox
出现如图,就是成功了
3.好不容易准备好数据,运行又开始报错
遇见这个问题不要慌
1.首先检测cuda是否安装
2.检测cuda版本与Paddle版本匹配
4.numpy又有问题
这个也不要慌,numpy版本不对
卸载重装即可!!!
5.开始练丹
单卡训练语句
python tools/train.py -c configs/yolov3/yolov3_mobilenet_v1_roadsign.yml
训练好以后开始验证,当然也可以边训练边验证
评估
python tools/eval.py -c configs/yolov3/yolov3_mobilenet_v1_roadsign.yml -o weights=output/yolov3_mobilenet_v1_roadsign/model_final
//这里注意weights路径要换成自己的,按照官方文档那样写的运行会报错
里面的参数有些还是没有看明白
验证完以后就可以找一张图片来预测了
预测语句
python tools/infer.py -c configs/yolov3/yolov3_mobilenet_v1_roadsign.yml --infer_img=demo/I102.png --output_dir=infer_output/ --draw_threshold=0.5 -o weights=output/yolov3_mobilenet_v1_roadsign/model_final --use_vdl=Ture
然后去结果保存路径查看预测结果
训练可视化加边训练边验证
python -m paddle.distributed.launch tools/train.py -c configs/yolov3/yolov3_mobilenet_v1_roadsign.yml --eval --use_vdl=true --vdl_log_dir=vdl_dir/scalar
可视化训练结果
visualdl --logdir vdl_dir/scalar/