算法进阶迁移学习加载数据集训练
算法进阶迁移学习加载数据集训练加载数据集class Image_Data(Dataset): def __init__(self,img_h=128,img_w=128,path,mode='train',process=True): self.img_h=img_h self.img_w=img_w self.path=path self.mode=mode self.process=process if self.mode is 'train': self.path=
