#!/usr/bin/env python3
# coding: utf-8
#生成trian和val的图片 随机抓取
import os
import random
output_path=r'E:\wapiandataset\JPEGImages'
list1=os.listdir(output_path)
for i in range(0,len(list1)):
#for imgpath in list1[i]:
imgpath=os.path.join(output_path,list1[i])
newlist=os.listdir(imgpath)
print(newlist)
print('------------')
random.shuffle(newlist)
val=int(len(newlist)*0.2)
print(val)
train = newlist[val:len(newlist)]
val=newlist[0:val]
print(val)
print(train)
with open("vall.txt", "a+") as f: # 打开文件
for j in range(0,len(val)):
f.write(str(val[j].split('.')[0]))
f.write('\n')
print('j',j)
with open("trainn.txt", "a+") as f: # 打开文件
for k in range(0, len(train)):
f.write(str(train[k].split('.')[0]))
f.write('\n')
print('k',k)
数据格式:大文件夹下多个小文件夹,小文件夹内存储图片