分类：加载自定义数据

2021
03-07

pytorch从csv加载自定义数据模板的操作

整理了一套模板，全注释了，这个难点终于克服了fromPILimportImageimportpandasaspdimportnumpyasnpimporttorchvision.transformsastransformsfromtorch.utils.dataimportDataset,DataLoaderimportos#放文件的路径dir_path='./97/train/'csv_path='./97/train.csv'classMydataset(Dataset):#传递数据路径，csv路径，数据增强方法def__init__(self,dir_path,csv,transform=None,target_transform=None):... 继续阅读 >