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.

相关推荐
zd2005727 分钟前
STREAMS指南:环境及宿主相关微生物组研究中的技术报告标准
人工智能·python·算法
Data_agent11 分钟前
京东商品价格历史信息API使用指南
java·大数据·前端·数据库·python
云技纵横12 分钟前
Stream API 从入门到实践:常用操作、易错点与性能建议
开发语言·windows·python
雪域迷影12 分钟前
macOS系统上或首次使用Python的urllib模块时出现 ssl.SSLCertVerificationError 错误
python·macos·ssl
其美杰布-富贵-李18 分钟前
PyTorch Optimizer 与 Scheduler 指南
人工智能·pytorch·python·优化·训练
Franklin40 分钟前
如何解决git HEAD detached 分离头指针问题
git·python·pycharm
deephub1 小时前
Anthropic 开源 Bloom:基于 LLM 的自动化行为评估框架
人工智能·python·自动化·大语言模型·行为评估
万兴丶1 小时前
批量转换音频格式工具
python
大学就业之家1 小时前
智能选岗的陷阱与应对策略
python