随机提取文件夹中百分比数量的文件(图片&对应标签)
自己制作数据集的话难免需要去验证标注是否准确,但验证时随机去除百分比数量的图片及标签即可。
import argparse
import os
import random
import shutil
from PIL import Image
def main(opt):
p = opt.percentage
img_dir = opt.image_dir
lab_dir = opt.label_dir
res_img_dir = opt.res_img_dir
res_lab_dir = opt.res_lab_dir
shutil.rmtree(res_img_dir) # 清空文件中内容
shutil.rmtree(res_lab_dir)
if not os.path.exists(res_img_dir): # 如果文件夹不存在,则创建文件夹
os