将图片打乱，生成训练集，测试集，比例8:2

#!/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)

数据格式：大文件夹下多个小文件夹，小文件夹内存储图片