ValueError: operands could not be broadcast together with shapes (1404,2000,1) (1404,2000)

ValueError: operands could not be broadcast together with shapes (1404,2000,1) (1404,2000)

在这里插入图片描述
错误原因:
此处图片为灰度图片,通道数为1,shape为(351,500),现在需要(351,500,1),所以报错

解决方法:
方法1:直接将灰度图片在数据集中去掉,只剩下多通道的数据,即,去掉shape为(351,500)的图片。此时程序正常运行,输出结果。
在这里插入图片描述
方法2:使用cv2将灰度单通道图转为多通道图,实际上还是个灰色图片,知识多了几个通道而已,每个通道都是灰色。
具体操作可参考以下链接:
转发链接:https://discuss.pytorch.org/t/runtimeerror-given-groups-1-weight-of-size-64-3-7-7-expected-input-3-1-224-224-to-have-3-channels-but-got-1-channels-instead/30153/4

方法3:本文是复现Real-ESRGAN,只需要加上两句代码即可。
在inference_realesrgan.py的140行后面加上:

        elif len(img.shape) == 2:
            img_mode = 'RGBA' 

修改后附近完整代码如下:

        img = cv2.imread(path, cv2.IMREAD_UNCHANGED)
        print(img.shape)
        if len(img.shape) == 3 and img.shape[2] == 4:
            img_mode = 'RGBA'
        elif len(img.shape) == 2:
            img_mode = 'RGBA'    #将灰度照片的单通道改为4通道
        else:
            img_mode = None

如下图所示,(23,236)的图片运行完后直接结束程序,说明正常处理,并且输出了结果。
在这里插入图片描述

  • 4
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
在Python中,当你尝试对两个数组进行操作时,如果它们的形状不兼容,就会出现"ValueError: operands could not be broadcast together with shapes"错误。这个错误通常发生在你尝试对形状不同的数组进行元素级别的操作时,比如加法、减法、乘法等。 要解决这个错误,你可以考虑以下几种方法: 1. 确保数组的形状相同:你可以使用NumPy库的reshape()函数来调整数组的形状,使其与另一个数组的形状相匹配。例如,如果你有一个形状为(2,)的数组和一个形状为(100,)的数组,你可以使用reshape()函数将第一个数组的形状调整为(2,1),然后再进行操作。 ```python import numpy as np array1 = np.array([1, 2]) array2 = np.array([1, 2, 3, ..., 100]) # 调整数组形状 array1_reshaped = array1.reshape((2, 1)) # 进行操作 result = array1_reshaped + array2 ``` 2. 使用广播功能:NumPy库提供了广播功能,可以自动调整数组的形状,使其与另一个数组的形状相匹配。你只需要确保数组的维度兼容即可。例如,如果你有一个形状为(2,)的数组和一个形状为(100,)的数组,你可以使用广播功能直接进行操作。 ```python import numpy as np array1 = np.array([1, 2]) array2 = np.array([1, 2, 3, ..., 100]) # 进行操作 result = array1 + array2 ``` 3. 检查数组的形状:在出现这个错误时,你可以检查一下两个数组的形状,确保它们是你期望的形状。你可以使用NumPy库的shape属性来获取数组的形状。 ```python import numpy as np array1 = np.array([1, 2]) array2 = np.array([1, 2, 3, ..., 100]) # 检查数组形状 print(array1.shape) # 输出:(2,) print(array2.shape) # 输出:(100,) ```

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

打史莱姆的小仙女~

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值