yolov4-pytorch训练自己的数据集_yolov4训练自己的数据集pycharm

作者：木道寻08 | 2024-07-24 13:42:51

踩

yolov4训练自己的数据集pycharm

第一步：

将图片和xml文件放入VOC2007文件夹中对应的目录
1

执行list.py文件生成txt
1

修改voc.py中的类别为自己的类别文件，执行voc.py生成train.txt文件
1

执行keans.py生成ahchor文件
修改model_data里面voc的类别中的类别标签为自己类别
1
2

修改train.py加载权重文件和训练文件的路径，并且根据是否有显卡设置Ture或者False
执行脚本开始训练
1
2

修改yolo.py文件中自己的图片路径加载和模型加载路径测试模型输出结果
1

在这里插入图片描述

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/木道寻08/article/detail/874996