关于在使用pytorch创建读取自己的分类数据集中出现
area = img.size[0] * img.size[1] TypeError: ‘int’ object is not subscriptable备忘
原因在于img数据类型不同,原始使用的是
image = cv2.imread(path)
由于
而transform需要的结构为
area = img.size[0] * img.size[1]
size由长宽组成,不适配
因此改用PIL读取图片
from PIL import Image
对应读取操作改为
image = Image.open(path)
解决问题