随机挑选分类训练集和测试集

# -*- coding: utf-8 -*-

import os
import random
import shutil
from shutil import copy2
from tqdm import tqdm

# category = ['可回收垃圾', '有害垃圾', '厨余垃圾', '其他垃圾']
category = ['0','1','2','3','4','5','6','7','8',