关于area = img.size[0] * img.size[1] TypeError: ‘int‘ object is not subscriptable 问题备忘

关于在使用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)

解决问题