学习笔记22-分类网络自动创建各自类别的测试集和训练集（代码已实现）

只需要创建一个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