PyTorch学习笔记(5)Dataloader与Dataset(2)

人民币识别 split_dataset import os import random import shutil def makedir(new_dir): if not os.path.exists(new_dir): os.makedirs(new_dir) if __name__ == '__main__': random.seed(1) dataset_dir = os.path.join(data, RMB_data) split_dir = os.path.join(data, rmb_split) trai