生成训练集、验证集、测试集的txt路径文件(适用于目标检测)

本文介绍了一个Python脚本,用于在指定目录下操作train、val、test文件夹,生成或重排包含图像文件路径的txt文件,实现数据集的随机打乱。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

 将代码脚本与train、val、test文件夹放在同一目录下,在该目录下打开终端运行脚本即可

import os
import sys
import random



dir_path = os.getcwd()
print(dir_path)

lst = ['test', 'train', 'val']


for i in lst :
    f_path = os.path.join(dir_path ,i)
    with open(f_path + '.txt', 'w') as ft:
        print(f_path + '.txt')

        for path, dir_lst, file_lst in os.walk(f_path):
            for file in file_lst:
                if os.path.splitext(file)[1] in ['.jpg', '.jpeg', '.png', '.JPG']:
                    root = os.path.join(path, file+'\n')
                    print(root)
                    ft.writelines(root)

                # 如果没有对应的txt文件
                txt_name = os.path.splitext(file)[0] + ".txt"
                txt = os.path.join(path, txt_name)
                if not os.path.exists(txt):
                    with open(txt, 'w') as t:
                        t.close()


# 打乱                    
os.system("shuf train.txt -o train.txt")
print('shuf train.txt -o train.txt')

os.system("shuf test.txt -o test.txt")
print('shuf test.txt -o test.txt')

os.system("shuf val.txt -o val.txt")
print('shuf val.txt -o val.txt')

 

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

lucky169

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值