赞
踩
从上面得到的images和labels数据还不能够直接训练,需要按照一定的比例划分训练集和验证集,并按照下面的结构来存放数据,划分代码如下所示,该部分内容和YOLOv8的训练、验证、预测及导出[目标检测实践篇]_yolov8训练测试验证-CSDN博客是重复的,代码如下:
- """
- 随机划分训练集和验证集
- """
- import os
- import random
- from shutil import copyfile
-
- # 输入路径
- image_dir = r'G:\yolov8\data_obb\images' # 替换成你的图像文件夹路径
- label_dir = r'G:\yolov8\data_obb\labels' # 替换成你的标签文件夹路径
-
- # 输出路径
- train_image_dir = r'G:\yolov8\ultralytics-main\ultralytics-main\my_data\obb\train\images'
- train_label_dir = r'G:\yolov8\ultralytics-main\ultralytics-main\my_data\obb\tr
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。