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

原因分析1.读取的是.png 图片是32位深度的，一般我们读取的是.jpg 24位深度的2.转换为RGB就行了test_image = Image.open(test_image_name).convert('RGB')

Golden-sun

7648人浏览 · 2020-09-10 20:06:12

Golden-sun · 2020-09-10 20:06:12 发布

原因分析

1.读取的是.png 图片是32位深度的，一般我们读取的是.jpg 24位深度的
2.转换为RGB就行了

test_image = Image.open(test_image_name).convert('RGB')

2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

[ 开源 ] FastAPI + LangGraph 实战智能客服 Agent：从工单分类到自动回复与业务回写附github

2048 AI社区

大厂Java面试实战：Spring Boot/Cloud、Redis/Kafka、JVM调优与Spring AI RAG（内容社区UGC+AIGC客服场景）

以“内容社区+UGC+AIGC智能客服”为业务背景，模拟大厂Java面试：严肃面试官循序渐进提问Spring Boot/Cloud、JVM、Redis、Kafka、数据库、观测、CI/CD与Spring AI/RAG/Agent等，小Y简单题会答复杂题含糊。文末给出每题详细答案与落地方案，便于小白系统学习。