使用飞浆训练目标检测模型

 参考链接:

PP-PicoDet算法训练行人检测模型-CSDN博客文章浏览阅读306次。PP-PicoDet模型特点:方案选择PP-PicoDet轻量化模型,主要看中PP-PicoDet体积小、速度快、精度较高的优势,非常适合本项目的部署环境和性能要求。同时,飞桨提供的预训练模型也可以最大程度上提升模型的收敛速度和精度。https://blog.csdn.net/qq_45437316/article/details/134519467

训练命令:

python tools/train.py  -c configs/picodet/legacy_model/picodet_s_320_voc.yml  --use_vdl=true --vdl_log_dir=vdl_dir/person_voc_log --eval>person_voc.log 2>&1&
 

导出模型命令:

python tools/export_model.py -c configs/picodet/legacy_model/picodet_s_320_voc.yml -o weights=output/picodet_s_320_voc/best_model.pdparams --output_dir=inference_model
 

推理命令:

python deploy/python/infer.py --model_dir=inference_model/picodet_s_320_voc --output=output/test --image_file=output/32.jpg --threshold=0.5 --device=GPU

你可能感兴趣的:(深度学习,机器学习,人工智能)