将数据集随机分为训练集和测试集
原数据保存形式:
处理后的数据保存形式:
代码:
# *_*coding: utf-8 *_*
# Author --LiMing--
import os
import random
import shutil
import time
def copyFile(fileDir, class_name):
image_list = os.listdir(fileDir) # 获取图片的原始路径
image_number = len(image_list)
train_number = int