【pytorch】关于OpenCV和PIL.Image读取图片的区别

CoCo_2022

于 2022-08-09 22:21:28 发布

阅读量1.9k

点赞数

分类专栏：机器学习文章标签： opencv pytorch python

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接：https://blog.csdn.net/cxh21627/article/details/126255953

版权

本文从数据使用方角度探讨PyTorch中使用OpenCV和PIL.Image读取图像的区别。PIL读取的图像形状为[W,H]，通道为RGB，而OpenCV读取的图像形状为[H,W,C]，通道为BGR。在转换到PyTorch模型输入时，图像需要转为PIL.Image格式并增加batch维度。在处理中文标签时，保存图像则推荐使用PIL。" 126604168,15242354,.NET 6与OpenCVSharp结合开发的WPF自用工具,"['计算机视觉', 'OpenCV', '.NET框架', 'WPF开发', '深度学习']

摘要由CSDN通过智能技术生成

数据使用方角度

首先从pytorch出发，torchvision.transforms()要求传入的图像是PIL.Image格式（通道要求是RGB格式的），另外模型处理输入要转换为[1,channel,H,W]；

所以最终导入torchvision.transforms()的图像格式需要转成PIL.Image，且需要在转换后增加batch维度([channel,H,W]变成[1,channel,H,W])


	test_transform=torchvision.transforms.Compose([
		...
	    transforms.ToTensor(),
	    ...
	])
	
	input_img=test_transform(img_pil).unsqueeze(0)
	pred_logits=model(input_img)

OpenCV和PIL.Image区别

1. PIL.Image

a. 读取图片


	from PIL import Image
	img_pil = Image

最低0.47元/天解锁文章

关注

0
点赞
踩
9

收藏

觉得还不错? 一键收藏
0
评论
【pytorch】关于OpenCV和PIL.Image读取图片的区别

首先从pytorch出发，torchvision.transforms()要求传入的图像是PIL.Image格式（通道要求是RGB格式的），另外模型处理输入要转换为[1,channel,H,W]；所以最终导入torchvision.transforms()的图像格式需要转成PIL.Image，且需要在转换后增加batch维度([channel,H,W]变成[1,channel,H,W])...
复制链接

扫一扫

专栏目录

CoCo_2022 CSDN认证博客专家 CSDN认证企业博客

码龄13年

75: 原创

3万+: 周排名

223万+: 总排名

10万+: 访问

: 等级

1008: 积分

1726: 粉丝

70: 获赞

65: 评论

509: 收藏

私信

关注

热门文章

分类专栏

机试题目 71篇
机器学习 4篇

最新评论

华为OD机试真题：组成最大数
wangyixin_92: nums = list(map(int, input().split(','))) def getMax(nums): result = '' if len(nums) > 25: return "输入的不能超过25个数字" for i in range(len(nums)): for j in range(len(nums)-1): if nums[j+1] > nums[j]: nums[j], nums[j+1] = nums[j+1], nums[j] for i in nums: result = result + str(i) return result result = getMax(nums) print(result)
华为OD机试真题：数据分类
weixin_42806016: 看懂了，谢谢佬
华为OD机试题：消除相邻且相同
MISAYAONE: 华为OD 2023 最新最全机试题库及讲解，100%通过率。 Java: https://renjie.blog.csdn.net/article/details/127947829 Python: https://renjie.blog.csdn.net/article/details/127946125 C++: https://renjie.blog.csdn.net/article/details/126965954 Js: https://renjie.blog.csdn.net/article/details/128974467 C: https://renjie.blog.csdn.net/article/details/129190260
华为OD机试真题：分糖果【2022 Q1 Q2 |200分】
MISAYAONE: 华为OD 2023 最新最全机试题库及讲解，100%通过率。 Java: https://renjie.blog.csdn.net/article/details/127947829 Python: https://renjie.blog.csdn.net/article/details/127946125 C++: https://renjie.blog.csdn.net/article/details/126965954 Js: https://renjie.blog.csdn.net/article/details/128974467 C: https://renjie.blog.csdn.net/article/details/129190260
华为OD机试题：统计射击比赛成绩
MISAYAONE: 华为OD 2023 最新最全机试题库及讲解，100%通过率。 Java: https://renjie.blog.csdn.net/article/details/127947829 Python: https://renjie.blog.csdn.net/article/details/127946125 C++: https://renjie.blog.csdn.net/article/details/126965954 Js: https://renjie.blog.csdn.net/article/details/128974467 C: https://renjie.blog.csdn.net/article/details/129190260

大家在看

最新文章

目录

评论

被折叠的条评论为什么被折叠?

到【灌水乐园】发言

查看更多评论

添加红包

成就一亿技术人!

hope_wisdom

发出的红包

实付元

使用余额支付

点击重新获取

扫码支付

钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。