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.

相关推荐
qq_393060473 分钟前
在wsl 上运行window的R代码
开发语言·python·r语言
@小码农4 分钟前
速查!2024 CSP-J/S第一轮认证成绩查询及晋级分数线
开发语言·c++·python·职场和发展·蓝桥杯·noi
weixin_3776348415 分钟前
图像阴暗度检测 逆光检测 背光检测
图像处理·python·计算机视觉
高垚淼20 分钟前
如何构建智能应用:深入探索Langchain的强大功能与应用潜力
人工智能·python·langchain
MavenTalk24 分钟前
Python在进行LLM应用相关开发常用的技术框架
开发语言·python·大模型·llm·大语言模型
openMind开发者1 小时前
Qwen2.5 PyTorch模型微调入门实战
人工智能·python
Goodness20201 小时前
STL与PLY格式转化
c++·python
Bruce_Liuxiaowei1 小时前
VSCode远程切换Python虚拟环境
ide·vscode·python
好奇的菜鸟2 小时前
Python之禅:编程哲学的精髓
开发语言·python