python 深度学习图片格式转换的总结

Soul·Top

于 2020-11-27 12:38:06 发布

阅读量700

点赞数

分类专栏：深度学习文章标签：计算机视觉 opencv

本文链接：https://blog.youkuaiyun.com/weixin_43869778/article/details/110227503

版权

本文总结了Python深度学习中图片格式的转换方法，包括cv2到tensor，cv2到numpy（PIL.Image），numpy（PIL.Image）回转cv2的操作，并详细介绍了如何将归一化的0-1图片恢复到0-255范围。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

cv2 转 tensor

img = cv2.imread("a.png")
# 转成 RGB 形式
img0 = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
# 归一化，并添加一个维度（Batch）
img0 = torch.from_numpy(img0).div(255.0).unsqueeze(0) # N*W*H*C
# 置换维度形式为 N*C*W*H
img = img0.permute(0, 3, 1,

最低0.47元/天解锁文章