在深度学习进行数据训练时,通常需要划分训练集、验证集和测试集,有一种做法就是不进行数据的划分,而使用.json格式的数据来指定哪些数据用来训练、哪些数据用于测试;
本代码用于生成json格式的文件,(仅供学习以及个人回忆)
import json
from os.path import join
import glob
import random
# path to folder that contains images
img_folder = './justPoint/big_image/'
# 随机挑选400张图片作为val
val_image = []
# ...此处省略将val_image填入内容代码...
random.shuffle(val_image) #乱序处理
with open('val.json','w') as f:
json.dump(val_image,f)
|