数据集创建:
文件夹:yolov7-main/VOC下面有
images存放原始图像;xml存放xml格式文件;此时labels和dataSet为空。
一:运行split_train_val.py,注意修改py文件中的文件地址。
二:运行voc_label.py文件。依然要注意修改文件夹地址。
三:最后查看生成在VOC目录下的几个txt中,图片的绝对路径是否正确!!!!!
此时数据集准备完毕。
打开train.py文件,修改训练轮数,batch_size等。
开始训练。
数据集创建:
文件夹:yolov7-main/VOC下面有
images存放原始图像;xml存放xml格式文件;此时labels和dataSet为空。
一:运行split_train_val.py,注意修改py文件中的文件地址。
二:运行voc_label.py文件。依然要注意修改文件夹地址。
三:最后查看生成在VOC目录下的几个txt中,图片的绝对路径是否正确!!!!!
此时数据集准备完毕。
打开train.py文件,修改训练轮数,batch_size等。
开始训练。