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.

相关推荐
superman超哥2 分钟前
仓颉代码内联策略深度解析
c语言·开发语言·c++·python·仓颉
写代码的【黑咖啡】6 分钟前
Python中的文件操作详解
java·前端·python
q_30238195567 分钟前
秒级筛查+94.7%精准!华为Atlas 200 DK边缘设备解锁糖尿病视网膜病变检测新范式
人工智能·python·深度学习·智能体
AndrewHZ14 分钟前
【python与生活】从手机定位到车辆导航:GPS定位算法原理与Python实现
python·智能手机·生活·gps·定位算法·北斗卫星·车辆导航
Edward.W32 分钟前
PyQt6 打造苹果风格 ADB 图形化工具:adbUI 深度测评与实战指南
python·adb·pyqt
纪伊路上盛名在37 分钟前
vscode的colab扩展目前的一些问题
ide·vscode·python·编辑器·colab·前后端
宁大小白40 分钟前
pythonstudy Day41
python·机器学习
盼哥PyAI实验室44 分钟前
Python 爬虫核心基础:请求与响应机制全解析(从 GET 请求到 JSON 分页实战)
爬虫·python·json
Tipriest_44 分钟前
Python 常用特殊变量与关键字详解
linux·python·关键字·特殊变量
Salt_07281 小时前
DAY 41 Dataset 和 Dataloader 类
python·算法·机器学习