【Python代码】随机抽取文件名列表NameList中的Name作为训练集

本文介绍了一种使用Python随机抽取指定数量图片名称的方法,并将其分为测试集与训练集两部分,通过这种方式可以有效地进行数据集的划分。

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

#!/usr/bin/env python
#coding=utf-8

#随机抽取一部分图片作为测试集


import random

NameList=[]#存储所有图片名字
'''
NameListPath='NameList_v.txt' #图片名存储地址
SlicePath='Main visibal/test.txt' #抽取的元素的存储地址
SlicePath2='Main visibal/trainval.txt' #剩余的元素存储地址
'''
NameListPath='Main visibal/trainval.txt'#图片名存储地址
SlicePath='Main visibal/val.txt' #抽取的元素的存储地址
SlicePath2='Main visibal/train.txt' #剩余的元素存储地址

#slice_num=164 #抽取数量
slice_num=66 #抽取数量

file_object = open(NameListPath)
try:
    for line in file_object:
        '''line_new=line[ :-2] #截掉Windows下的\r\n
        line_new+='\n' #增加换行
        NameList.append(line_new)'''
        NameList.append(line)
finally:
    file_object.close( )

#随机抽取
slice = random.sample(NameList, slice_num) 
#print slice

#存储随机抽取的元素
output1 = open(SlicePath, 'w')
try:
    output1.writelines(slice)
finally:
    output1.close()

#抽取后剩余
slice2=NameList
for name in slice:
    slice2.remove(name)
#print slice2

#存储随机抽取后剩余的元素
output2 = open(SlicePath2, 'w')
try:
    output2.writelines(slice2)
finally:
    output2.close()
    
print 'Done!'

 

转载于:https://www.cnblogs.com/xiangfeidemengzhu/p/7058236.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值