【深度学习】图片张量错误 The size of tensor a (4) must match the size of tensor b (3) at non-singleton

Pytorch 写 Dataset 的时候遇到的这个问题The size of tensor a (4) must match the size of tensor b (3) at non-singleton解决方法Image.open(image)改成Image.open(image).convert("RGB")

MangoloD

811人浏览 · 2021-03-28 20:02:10

MangoloD · 2021-03-28 20:02:10 发布

Pytorch 写 Dataset 的时候遇到的这个问题

The size of tensor a (4) must match the size of tensor b (3) at non-singleton

解决方法

Image.open(image)

改成

Image.open(image).convert("RGB")

DAMO开发者矩阵

DAMO开发者矩阵，由阿里巴巴达摩院和中国互联网协会联合发起，致力于探讨最前沿的技术趋势与应用成果，搭建高质量的交流与分享平台，推动技术创新与产业应用链接，围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐

【机器人避障】基于全自主差动驱动移动机器人复杂环境中动态路径跟踪和实时障碍物规避附Matlab代码

一、开篇引入：从需求到核心 —— 差动驱动机器人的自主导航难题一、开篇引入：从需求到核心 —— 差动驱动机器人的自主导航难题1.1 应用场景驱动：差动驱动机器人的 “用武之地”在科技飞速发展的当下，全自主差动驱动移动机器人凭借独特的结构与灵活的运动能力，在众多领域大显身手。在军事侦察领域，它能深入危险区域，凭借小巧身形穿梭于复杂地形，为作战指挥提供关键情报，降低士兵暴露于危险环境的风险；工业物料

DAMO开发者矩阵

【图像加密】基于仿射变换和双随机相位的图像加密算法实现附matlab代码

在大数据传输、云端存储、军事通信、医疗影像共享等场景中，图像作为核心信息载体，其隐私保护至关重要。传统明文传输或简单加密方式易遭窃取、篡改，而图像的冗余性、相关性等特性，也对加密算法提出了 “高安全性、抗攻击、易实现” 的核心要求 —— 图像加密的本质，是通过数学变换将原始图像转化为无意义的乱码，仅授权者可通过密钥还原，从而保障信息安全。