import os
import random
import hashlib
f1 = open("train2.txt","w")
f2 = open("val2.txt","w")
lst = os.listdir("./")
random.shuffle(lst)
for l in lst:
a = hashlib.md5(l.encode("utf-8"))
b = a.hexdigest()
as_int = int(b, 16)
id = as_int%10
if id<=8:
f1.write(l+" "+str(as_int)+"\n")
else:
f2.write(l+" "+str(as_int)+"\n")
f1.close()
f2.close()
split train val dataset
最新推荐文章于 2024-06-23 12:08:35 发布