深度学习-图片分类-数据预处理

Question

现有数据：图片，train.csv

train.csv包括两列：图片文件名，label

工具：pytorch

疑惑：

写自定义dataset时，通过读取csv文件，将name list和label list读取保存，在get item方法中根据idx读取相应的图片，并裁剪，to tensor，但是label并未做任何处理并返回tensor和label，那么在此时是否需要对label进行处理比如：get dummy 或者其他？

遇到问题：在进行训练时，mse loss 报错：AttributeError: 'tuple' object has no attribute 'size'

 

Answer 1

label也要放到tensor中，训练中出现问题应该是你那个变量不是tensor，导致你用。size方法就找不到啦