用img = cv2.imread(path)加载数据集后,在进行transform时报错:TypeError: Unexpected type
transform = transforms.Compose([
transforms.Resize((112, 112)),
transforms.RandomHorizontalFlip(),
transforms.ToTensor(), # range [0, 255] -> [0.0,1.0]
transforms.Normalize(mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5))
])
报错原因:这是数据格式问题:把transforms.Resize((112, 112))去掉,直接在img = cv2.imread(path)后进行img = cv2.resize(img, (112, 112))
精彩文章
发表评论