将数据集分为训练集和验证集存放在新的文件夹中
import os
import random
from PIL import Image
file_name = os.listdir('data/widerface/train/images') # file_name是个列表,用来装images目录下的文件名
if not os.path.exists('image'):
os.mkdir('image') # 创建一个我们要存放分类好的数据文件夹
s = ['train','val']
for j in s:
for i .
转载
2020-11-02 16:10:15 ·
1024 阅读 ·
0 评论