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.

相关推荐
MediaTea16 小时前
Python:实例 __dict__ 详解
java·linux·前端·数据库·python
SunnyDays101116 小时前
Python Excel 打印设置全攻略(打印区域、缩放、页边距、页眉页脚等)
python·excel打印设置·excel页面设置·excel打印选项
小鸡吃米…16 小时前
Python的人工智能-机器学习
人工智能·python·机器学习
傻啦嘿哟17 小时前
Python上下文管理器:优雅处理资源释放的魔法工具
开发语言·python
阿方索17 小时前
Python 基础简介
开发语言·python
BBB努力学习程序设计17 小时前
Python异步编程完全指南:从asyncio到高性能应用
python·pycharm
deephub17 小时前
机器学习时间特征处理:循环编码(Cyclical Encoding)与其在预测模型中的应用
人工智能·python·机器学习·特征工程·时间序列
追光天使17 小时前
Python 连接数据库并遍历数据
python
BBB努力学习程序设计17 小时前
Python迭代器与生成器深度解析:懒加载的艺术
python·pycharm
dazzle17 小时前
OpenCV基础教学(二):图像的灰度化处理
python·opencv·计算机视觉