yolov5训练自己的数据集

Leraine

于 2024-04-02 10:26:54 发布

阅读量570

点赞数 3

分类专栏：目标检测文章标签： YOLO 目标检测

本文链接：https://blog.youkuaiyun.com/weixin_43082603/article/details/129754522

版权

labelImg标记图片时选择yolo格式，保存为txt格式文件，若为xml格式使用时需要转换成txt格式。
在这里插入图片描述

在yolov5代码中新建VOCData文件夹。
在这里插入图片描述
images中存放需要检测的原始图片。labels中存放标记后的txt格式文件。

split_train_val.py

# 划分训练集
# coding:utf-8

import os
import random
import argparse

parser = argparse.ArgumentParser()
#标记后文件的地址，根据自己的数据进行修改
parser.add_argument('--input_path', default='E:\study\yolo\yolov5-pig20230404\VOCData\labels', type=str, help='input txt label path')
#数据集的划分，地址选择自己数据下的dataSet_path
parser.add_argument('--output_path', default='E:\study\yolo\yolov5-pig20230404\VOCData\dataSet_path', type=str, help='output txt label path'