[经验分享] 【玄铁杯第三届RISC-V应用创新大赛】目标检测算法yolofastest模型训练

wakojosin   2023-11-16 13:40 楼主
人体检测算法模型训练
  1. 数据准备
    下载mscoco数据集,使用关键点标签数据,并将只出现手脚,等极少部位人体的图像和标签过滤,下载aichallenge,将mscoco数据集和aichallenge数据集转换为yolo格式。
  2. 训练准备
    下载模型算法,我们选用yolofastestV2模型,下载指令:git clone https://github.com/dog-qiuqiu/Yolo-FastestV2.git
    运行python genanchors.py 聚类生成anchors
    134019c9oaqio9owipwpqt.png
  3. 模型训练
配置coco.data 文件为自己的路径
配置coco.name类别为person
训练:python train.py
134019sneoeas93cchqczj.png

 

回复评论 (2)

支持楼主的极力推荐,感谢楼主的无私奉献,顶起

点赞  2023-11-16 15:12
老哥,你模型用的是几啊?我用原版报错捏
点赞  2023-11-17 11:30
电子工程世界版权所有 京B2-20211791 京ICP备10001474号-1 京公网安备 11010802033920号
    写回复