加载数据集报错TypeError: Unexpected type ＜class ‘numpy.ndarray‘＞

原创已于 2024-04-02 23:11:26 修改 · 5.4k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #神经网络

于 2023-09-20 23:51:55 首次发布

深度学习543 专栏收录该内容

8 篇文章

订阅专栏

当使用cv2.imread加载图像并遇到TypeError时，原因是img数据类型为numpy.ndarray。解决方法是先通过transforms.ToTensor将其转换为Tensor格式，再进行如Resize和HorizontalFlip等transform操作。对于Image.open加载的图片，需先转为numpy.array或torch.tensor。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

    用img = cv2.imread(path)加载数据集后，在进行transform时报错：TypeError: Unexpected type <class 'numpy.ndarray'>

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))

补充：

cv2.imread(path)读取的图片是'numpy.ndarray'格式，要在pytorch里变化形状需要先用transforms.ToTensor()把格式转为tensor格式，才能 transforms.Resize((112, 112))与transforms.RandomHorizontalFlip(),

Image.open(path)读取的图片是PLI格式，需要转numpy.array()或torch.tensor()转格式才能进行相应的数值操作或者张量操作