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.

相关推荐
仙人掌_lz17 分钟前
深入理解深度Q网络DQN:基于python从零实现
python·算法·强化学习·dqn·rl
小雅痞1 小时前
[Java][Leetcode middle] 80. 删除有序数组中的重复项 II
java·python·leetcode
大叔_爱编程1 小时前
p020基于Django的4S店客户管理系统
vue.js·python·django·毕业设计·源码·课程设计·4s店客户管理系统
yorushika_1 小时前
python打卡训练营打卡记录day22
开发语言·python·机器学习
代码的乐趣1 小时前
支持selenium的chrome driver更新到136.0.7103.92
chrome·python·selenium
带鱼工作室1 小时前
通义读光系列文字检测+识别模型端到端OCR应用
python·opencv·计算机视觉·ocr
安特尼1 小时前
招行数字金融挑战赛数据分析赛带赛题二
python·算法·机器学习·金融·数据分析
Tom Boom1 小时前
14. 原生测试框架Unittest的skip、skipIf、skipUnless的使用
自动化测试·python·测试开发·unittest·自动化测试框架开发
小白爱编程HC2 小时前
用pymysql操作数据库
数据库·python·mysql
尸僵打怪兽3 小时前
软考错题集
java·python·计算机网络·操作系统·c·软考·计算机组成原理