图片类型

pytorch 实现将自己的图片数据处理成可以训练的图片类型

为了使用自己的图像数据，需要仿照pytorch数据输入创建新的类，其中数据格式为numpy.ndarray。 将自己的图片保存到numpy.ndarray中，然后创建类 from torch.ut