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

wakojosin

【玄铁杯第三届RISC-V应用创新大赛】目标检测算法yolofastest模型训练 [复制链接]

 

人体检测算法模型训练

1. 数据准备
   下载mscoco数据集，使用关键点标签数据，并将只出现手脚，等极少部位人体的图像和标签过滤，下载aichallenge，将mscoco数据集和aichallenge数据集转换为yolo格式。
2. 训练准备
   下载模型算法，我们选用yolofastestV2模型，下载指令：git clone
   链接已隐藏，如需查看请登录或者注册
   
   运行python genanchors.py 聚类生成anchors
   
3. 模型训练

配置coco.data 文件为自己的路径

配置coco.name类别为person

训练：python train.py

 

chejm

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

吾妻思萌

老哥，你模型用的是几啊？我用原版报错捏