ValueError: pic should not have > 4 channels. Got XXX channels.

to_pil_image的channel设置在不同版本发生了重大改变, 是出现此错误的一个可能原因

这是一个torchvision.transforms.functional之下的function。在torchvision的0.16版本和0.17版本之间,其发生了重大改变。

python 复制代码
#0.16 版本
if pic.shape[-3] > 4:
        raise ValueError(f"pic should not have > 4 channels. Got {pic.shape[-1]} channels.")
#0.17版本
if pic.shape[-1] > 4:
        raise ValueError(f"pic should not have > 4 channels. Got {pic.shape[-1]} channels.")

例如segment_anything包中的ResizeLongestSide.apply_image就使用了这个函数,但是其安装要求没有强制要求torchvision>=0.17.

相关推荐
xier_ran7 分钟前
Python从入门到精通:(2)Python 核心进阶教程从数据结构到面向对象
linux·windows·python·microsoft
程序员三藏28 分钟前
接口自动化测试框架搭建详解
自动化测试·软件测试·python·测试工具·职场和发展·测试用例·接口测试
skywalk816339 分钟前
老显卡老cpu用vllm推理大模型失败Intel(R) Xeon(R) CPU E5-2643 v2
人工智能·pytorch·python·vllm
程序员爱钓鱼1 小时前
Python编程实战:文件读写(文本/二进制)详解与实战
后端·python·ipython
百锦再2 小时前
第6章 结构体与方法
android·java·c++·python·rust·go
TheInk2 小时前
python学习笔记之Python基础教程(crossin全60课)
笔记·python·学习
尘缘浮梦2 小时前
RobotFramework框架环境搭建
linux·开发语言·python
程序员爱钓鱼2 小时前
Python编程实战:try...except...finally —— 让程序更稳健的异常处理机制
后端·python
岁岁岁平安2 小时前
python MongoDB 基础
数据库·python·mongodb
闲人编程2 小时前
用Python控制硬件:Raspberry Pi项目初体验
开发语言·python·raspberry·pi·codecapsule·控制硬件