python实现yolo打乱数据集分配到train.txt 和val.txt代码

yolo打乱数据集分配到train.txt 和val.txt代码

import os
import numpy as np
import random
image_list=[]
for i in range(1,5505):
    image_list.append(str(i).zfill(8))
fv=open('val.txt','a+')
ft=open('train.txt','a+')
print(len(image_list))
val=[]
val=random.sample(image_list,500)
train=[]
for item in image_list:
    if item in val:
        pass
    else:
        train.append(item)
print(len(train))
train=random.sample(train,5004)

for i in val:
    fv.write(i+'.jpg'+'\n')

for i in train:
    ft.write(i+'.jpg'+'\n')