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.

相关推荐
Want5954 小时前
Python跳动的爱心②
开发语言·python·数学建模
田里的水稻4 小时前
PyTorch_TensorBoard展示
人工智能·pytorch·python·深度学习·神经网络
张张123y4 小时前
后端第九课:部署上线(把你的 FastAPI 后端真正跑起来给别人用)
python·docker·部署·fastapi
云勤通4 小时前
智慧后勤系统哪家定制能力强
大数据·python
想七想八不如114084 小时前
元学习简单入门
人工智能·pytorch·python·深度学习·学习·机器学习
Zucker n4 小时前
一款适合研究生科研工作安排的日历
linux·数据库·python
尼给路达哟5 小时前
node(koa)和python(fastapi)面试
python·面试·fastapi
曲径海盗6 小时前
【Python 学习第 32 天】字符串基础
python·学习笔记·编程入门
田里的水稻7 小时前
PyTorch_torchvision---数据集下载\TensorBoard可视化
人工智能·pytorch·python·深度学习·机器学习
程序员正茂7 小时前
Win11使用自定义训练集训练ncnn模型
python·yolo·图像识别·移动端·ncnn