YOLOv8-第Y7周：训练自己的数据集

• 🍨 本文为🔗365天深度学习训练营 中的学习记录博客
• 🍖 原作者：K同学啊 | 接辅导、项目定制

一、配置环境

1. 官网下载地址

官网地址：【YOLOv8开源地址】

2.安装需要的环境

安装环境依赖包,请进入项目文件在文件地址栏中直接输入cmd,打开cmd后输入如何打开CMD命令行_网络用户命令行工具在哪-CSDN博客

二、准备自己的数据集

 1、标签转换

XML—TXT

import xml.etree.ElementTree as ET
import os

voc_folder = r"F:/person-car/annotations/"  # 储存voc格式的标注文件的文件夹
yolo_folder = r"F:/person-car/txt/"  # 转换后的yolo格式标注文件的储存文件夹

class_id = ["person","car"]

# voc标注的目标框坐标值转换到yolo标注的目标框坐标值的函数
def convert(size, box):
    dw = 1. / size[0]
    dh = 1. / size[1]
    x = (box[0] + box[1]) / 2.0
    y = (box[2] + box[3]) / 2.0
    w = box[1] - box[0]
    h = box[3] - box[2]
    x = x * dw
    w = w * dw
    y = y * dh
    h = h * dh
    return (x, y, w, h)


# 对单个voc标注文件进行转换成其对应的yolo文件的函数
def convert_annotation(xml_file):
    file_name = xml_file.rstrip(".xml")  # 这一步将所有voc格式标注文件取出后缀名“.xml”，方便接下来作为yolo格式标注文件的名称
    print(file_name)
    in_file = open(os.path.join(voc_folder, xml_file),encoding='gb18030', errors='ignore')  # 打开当前转换的voc标注文件
    out_file = open(os.path.join(yolo_folder, file_name + ".txt", ), 'w',encoding='gb18030', errors='ignore')  # 创建并打开要转换成的yolo格式标注文件
    tree = ET.parse(in_file)
    root = tree.getroot()
    size = root.find('size')
    w = int(size.find('width').text)
    h = int(size.find('height').text)
    for obj in root.iter('object'):
        cls = obj.find('name').text
        cls_id = class_id.index(cls)
        print(cls_id)
        print(obj)
        xmlbox = obj.find('bndbox')
        b = (float(xmlbox.find('xmin').text),
             float(xmlbox.find('xmax').text),
             float(xmlbox.find('ymin').text),
             float(xmlbox.find('ymax').text))
        bb = convert((w, h), b)
        out_file.write(str(cls_id) + " " + " ".join([str(a) for a in bb]) + '\n')


xml_fileList = os.listdir(voc_folder)  # 将所有voc格式的标注文件的名称取出存放到列表xml_fileList中
for xml_file in xml_fileList:
    # 这里的for循环开始依次对所有voc格式标注文件生成其对应的yolo格式的标注文件
    print(xml_file)
    convert_annotation(xml_file)

2、数据集划分

训练集：验证集=8:2

import os
import random
import shutil


def moveFile(Txtdir):
    pathDir = os.listdir(Txtdir)  # 获取标签列表
    filenumber = len(pathDir)
    rate = 0.2  # 定义抽取图片的比例
    random_imgnum = int(filenumber * rate)  # 按照比例从文件夹中取一定数量标签
    sample = random.sample(pathDir, random_imgnum)  # 随机选取random_imgnum数量的样本标签
    print(sample)
    for labelname in sample:
        nameimg = os.path.splitext(labelname)[0]
        shutil.move(Txtdir + labelname, Txt_target_dir + labelname)
        shutil.move(Imgdir + nameimg + '.png', img_target_dir + nameimg + '.png')

    '