YOLO v7数据集格式 (附代码)
博主个人的数据集格式为 voc 数据集，由于之前转v5的时候是用 voc 数据集先转成YOLO格式数据集，这里主要是对YOLOv5数据集进行转换。
·
博主个人的数据集格式为 voc 数据集，由于之前转v5的时候是用 voc 数据集先转成YOLO格式数据集，这里主要是对YOLOv5数据集进行转换。
images 里的内容：
转换代码：
import glob
#存放图片的地址
train_image_path = r"D:\YOLOv7_Maize\images\train"
valid_image_path = r"D:\YOLOv7_Maize\images\val"
#生成的txt的路径
txt_path = r"D:\YOLOv7_Maize/"
def generate_train_and_val(image_path, txt_file):
with open(txt_file, 'w') as tf:
for jpg_file in glob.glob(image_path + '//' + '*.jpg'):
tf.write(jpg_file + '\n')
generate_train_and_val(train_image_path, txt_path + 'train.txt')
generate_train_and_val(valid_image_path, txt_path + 'valid.txt')
魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。
更多推荐
3
0- 0
已为社区贡献1条内容
所有评论(0)