只需要创建一个original文件夹,把图片放里面就ok,每个类别的train和val文件夹自动生成。
#自动分train和val文件
import os
from shutil import copy, rmtree
import random
def mk_file(file_path: str):
if os.path.exists(file_path):
# 如果文件夹存在,则先删除原文件夹在重新创建
rmtree(file_path)
os.makedirs(file_path)
def main():