博客

Yolo-fastestv2训练自己的数据集记录

第一节：代码来源

本机环境：ubuntu20，cuda,cudnn,pytorch==1.11.0
代码来源：https://github.com/dog-qiuqiu/Yolo-FastestV2
配置环境后先测试一下环境
终端输入：

python3 test.py --data data/coco.data --weights modelzoo/coco2017-0.241078ap-model.pth --img img/000139.jpg

在代码主目录下出现检测结果即可。

环境测试通过后，接下来需要准备数据集了。

在github上看到需要调整数据结构为该项目要求的格式。

第二节：数据处理

1.获取数据集
使用labelImg标注软件，选择yolo格式进行标注。
2.新建train和val文件夹
在data文件夹下，新建train和val文件夹，并将图片和txt标签置于对应文件夹中。

3.新建py文件
在主目录下新建sum_txt.py文件，写入：

import os
txt_ = r"/home/cj/work/CJ/Yolo-FastestV2-main/data/val.txt"  # 生成txt的名字
Imgdir="/home/cj/work/CJ/Yolo-FastestV2-main/data/val/"
pathDir = os.listdir(Imgdir)    #获取图像列表
with open(txt_, 'a+') as f:
    for i in pathDir:
        if os.path.splitext(i)[1] == '.jpg':
            item = (Imgdir + "{}").format(str(i).zfill(4))
            f.write(item)
            f.write("\n")

其中txt_ = r”/home/cj/work/CJ/Yolo-FastestV2-main/data/train.txt”这一行为你需要生成文件路径，需要获取train.txt和val.txt两个文件，对应是官网中的

第三节：训练

1.生成anchors
终端运行

python3 genanchors.py --traintxt ./data/train.txt

获得anchors。复制粘贴到coco.data。
2.修改coco.name和coco.data文件
coco.names中修改类别为自己的数据类别。
coco.data修改类别数量和anchors，epoch等

[name]
model_name=coco

[train-configure]
epochs=60
steps=150,250
batch_size=32
subdivisions=1
learning_rate=0.001

[model-configure]
pre_weights=None
classes=4
width=352
height=352
anchor_num=3
anchors=12.79186635,34.99124329, 17.43634664,51.82827546, 18.69456021,55.36147158, 40.11064084,29.4783456, 44.85245013,34.27625813, 44.92398179,31.9961749

[data-configure]
train=/home/cj/work/CJ/Yolo-FastestV2-main/data/train.txt
val=/home/cj/work/CJ/Yolo-FastestV2-main/data/val.txt
names=./data/coco.names

3.训练
终端输入

python3 train.py --data data/coco.data

第四节：测试模型
在训练结束后主目录下会有一个weight文件夹，里面是生成的权重文件。
输入测试指令即可

python3 test.py --data data/coco.data --weights modelzoo/coco2017-0.241078ap-model.pth --img img/000139.jpg

权重和测试图像要替换为自己的