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.

相关推荐
weixin_462901977 小时前
ESP32电压显示
开发语言·javascript·css·python
阿贵---7 小时前
使用Python进行PDF文件的处理与操作
jvm·数据库·python
winfredzhang7 小时前
FolderMover Pro用 Python + wxPython 构建安全高速的文件移动工具
python·wxpython·多线程·md5校验·容量筛选·失败回滚·实时进度
551只玄猫7 小时前
【基于python的金融分析和风险管理 学习笔记】中阶篇 第6章 分析利率和汇率
笔记·python·学习·金融·学习笔记·汇率·利率
小邓睡不饱耶7 小时前
东方财富网股票数据爬取实战:从接口分析到数据存储
开发语言·爬虫·python·网络爬虫
2401_891655817 小时前
GitHub镜像站搭建全攻略技术文章大纲
python·github
cm6543207 小时前
使用XGBoost赢得Kaggle比赛
jvm·数据库·python
yzx9910137 小时前
WorkBuddy 使用指南:解锁几大核心功能,提升工作效率
人工智能·python
蛐蛐蛐7 小时前
在昇腾310P推理服务器上安装CANN和PyTorch
人工智能·pytorch·python·npu
qq_416018727 小时前
游戏与图形界面(GUI)
jvm·数据库·python