【报错】如下:
RuntimeError: stack expects each tensor to be equal size, but got [3,256, 341] at entry 0 and [3, 256, 386] at entry 1
【分析】
train_dataset = datasets.ImageFolder(
traindir,
transforms.Compose([
transforms.Resize((224)) ###
【原因】
原因是transforms.Resize() 的参数设置问题,改为如下设置就可以了
【解决】
train_dataset = datasets.ImageFolder(
traindir,
transforms.Compose([
transforms.Resize((224,224)),
搞定!撒花…