1.旋转目标检测数据集划分和配置
从上面得到的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