RuntimeError: Given groups=1, weight of size [64, 3, 6, 6], expected input[1, 4, 896, 1280] to have

最新推荐文章于 2025-05-08 11:48:14 发布

随心而动的动

最新推荐文章于 2025-05-08 11:48:14 发布

阅读量3.3k

点赞数 1

CC 4.0 BY-SA版权

分类专栏：图片处理文章标签： python 视频编解码

本文链接：https://blog.csdn.net/USLL_7263/article/details/124128848

图片处理专栏收录该内容

1 篇文章

订阅专栏

本文详细介绍了在图片Base64编码解码时遇到通道数不匹配的问题，通过示例代码展示了如何使用Python将4通道图片转换为3通道RGB格式，确保图片在后续应用中的正确显示。

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

在使用图片base64编码时遇到RuntimeError: Given groups=1, weight of size [64, 3, 6, 6], expected input[1, 4, 896, 1280] to have 3 channels, but got 4 channels instead 问题

解决方法

该问题出现在png格式的图片base64编码后解码为数组格式时会出现通道数为4的情况，需要将解码的数组用函数convert(‘RGB’)做RGB的转换现在将代码给出如下：

// 将png编码的base64字符串解码为数组格式
def imgStr2imgArr(img_str):
    '''
    img_str： base64图片字符串
    '''
    buff = BytesIO(base64.b64decode(img_str))
    image = Image.open(buff).convert('RGB')
    img_arr = np.array(image)
    return img_arr