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.

相关推荐
Dontla2 小时前
用pip install -e .开发Python包时,Python项目目录结构(项目结构)(可编辑安装editable install)
python·pip
Thomas.Sir2 小时前
第三章:Python3 之 字符串
开发语言·python·字符串·string
威联通网络存储3 小时前
告别掉帧与素材损毁:威联通 QuTS hero 如何重塑影视后期协同工作流
前端·网络·人工智能·python
Dxy12393102163 小时前
Python 根据列表中某字段排序:从基础到进阶
开发语言·windows·python
splage3 小时前
Java进阶——IO 流
java·开发语言·python
cliffordl3 小时前
设计模式(python)
python·设计模式
always_TT4 小时前
从Python_Java转学C语言需要注意什么?
java·c语言·python
2301_793804694 小时前
定时任务专家:Python Schedule库使用指南
jvm·数据库·python
穿越世纪的风尘5 小时前
【问题解决】No module named ‘_sqlite3‘
python·centos