[经验分享]
【玄铁杯第三届RISC-V应用创新大赛】目标检测算法yolofastest模型训练
人体检测算法模型训练
- 数据准备
下载mscoco数据集,使用关键点标签数据,并将只出现手脚,等极少部位人体的图像和标签过滤,下载aichallenge,将mscoco数据集和aichallenge数据集转换为yolo格式。
- 训练准备
下载模型算法,我们选用yolofastestV2模型,下载指令:git clone https://github.com/dog-qiuqiu/Yolo-FastestV2.git
运行python genanchors.py 聚类生成anchors
- 模型训练
配置coco.data 文件为自己的路径
配置coco.name类别为person
训练:python train.py