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.

相关推荐
做怪小疯子16 小时前
华为笔试0429
python·numpy
Warson_L16 小时前
Dictionary
python
寒山李白18 小时前
解决 python-docx 生成的 Word 文档打开时弹出“无法读取内容“警告
python·word·wps·文档·docx·qoder
2401_8323655219 小时前
JavaScript中rest参数(...args)取代arguments的优势
jvm·数据库·python
Sirius.z19 小时前
第J3周:DenseNet121算法详解
python
2301_7796224119 小时前
Go语言怎么用信号量控制并发_Go语言semaphore信号量教程【入门】
jvm·数据库·python
2301_7662834419 小时前
c++如何将控制台输出保存到文件_cout重定向到txt【详解】
jvm·数据库·python
小康小小涵21 小时前
基于ESP32S3实现无人机RID模块底层源码编译
linux·开发语言·python
lzjava202421 小时前
Python的函数
开发语言·python
Awesome Baron1 天前
skill、tool calling、MCP区别
开发语言·人工智能·python