训练yolo 数据转换代码

最新推荐文章于 2024-08-26 21:08:43 发布

东哥爱编程

最新推荐文章于 2024-08-26 21:08:43 发布

阅读量109

点赞数

分类专栏： ai模型搭建，平台部署课程深度学习 ubuntu 文章标签： YOLO python 深度学习

本文链接：https://blog.csdn.net/monk96/article/details/129497574

版权

ai模型搭建，平台部署课程同时被 3 个专栏收录

33 篇文章 20 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

深度学习

31 篇文章 3 订阅

订阅专栏

ubuntu

11 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何使用Python进行YOLO数据集的分割、标签生成，并验证转换的正确性，详细解析了train_val_split.py脚本的执行过程。

摘要由CSDN通过智能技术生成

分割数据

train_val_split.py

# coding:utf-8
 
import os
import random
import argparse
 
parser = argparse.ArgumentParser()
# xml文件的地址，根据自己的数据进行修改 xml一般存放在Annotations下
parser.add_argument('--xml_path', default='/data/carData/', type=str, help='input xml label path')
#数据集的划分，地址选择自己数据下的ImageSets/Main
parser.add_argument('--txt_path', default='/data/train/', type=str, help='output txt label path')
opt = parser.parse_args()
project_path = os.getcwd() # 项目运行目录
# 8:1:1

# 4012
# 2081 /jpg
# 1931 /json
# 数据丢失
trainval_percent = 0.9
train_percent = 0.8
xmlfilepath = project_path + opt.xml_path
txtsavepath = project_path + opt.txt_path
total

了解本专栏