解决:RuntimeError: shape ‘[1, 3, 32, 32]‘ is invalid for input of size 4096

报错:

原因:

因为我下载的图片是4通道的:通过下面打印出来的结果可知是RGBA四通道的。

解决方法:

但是网络的输入需要RGB三通道,因此需要使用下列方法进行转换:

image_path = "../imgs/dog.png"
image = Image.open(image_path)
image = image.convert("RGB")
print(image) # size=455x297

可以发现转换后的格式为:

相关推荐

  1. openssl3.2/test/certs - 043 - 4096-bit leaf key

    2024-02-16 10:50:01       38 阅读
  2. springboot请求406、500问题

    2024-02-16 10:50:01       38 阅读
  3. Windows + RTX4090驱动,CUDA安装

    2024-02-16 10:50:01       29 阅读

最近更新

  1. TCP协议是安全的吗?

    2024-02-16 10:50:01       18 阅读
  2. 阿里云服务器执行yum,一直下载docker-ce-stable失败

    2024-02-16 10:50:01       19 阅读
  3. 【Python教程】压缩PDF文件大小

    2024-02-16 10:50:01       18 阅读
  4. 通过文章id递归查询所有评论(xml)

    2024-02-16 10:50:01       20 阅读

热门阅读

  1. 内网穿透与搭建私人服务器

    2024-02-16 10:50:01       30 阅读
  2. 如何使用AIGC才能有利于创新能力的培养

    2024-02-16 10:50:01       28 阅读
  3. Rust Option类型详解

    2024-02-16 10:50:01       27 阅读
  4. Rust 初体验4

    2024-02-16 10:50:01       32 阅读
  5. 如何使用 Python 通过代码创建图表

    2024-02-16 10:50:01       28 阅读
  6. Day35 贪心算法part04

    2024-02-16 10:50:01       29 阅读
  7. Spring boot整合redisson报错

    2024-02-16 10:50:01       32 阅读
  8. idea基础配置

    2024-02-16 10:50:01       39 阅读
  9. ArrayList 与 LinkedList 区别

    2024-02-16 10:50:01       32 阅读
  10. Rust 原生类型

    2024-02-16 10:50:01       31 阅读
  11. SP1:基于Plonky3构建的zkVM

    2024-02-16 10:50:01       37 阅读
  12. 软件设计原则

    2024-02-16 10:50:01       26 阅读
  13. uniapp中打开蓝牙需要哪些权限

    2024-02-16 10:50:01       28 阅读