当前位置: 首页 > news >正文

PIL、cv2、numpy,和pytorch(torch)之间的转换

在图像处理和深度学习中,经常需要在PIL(Python Imaging Library)、OpenCV(cv2)、NumPy和PyTorch之间进行图像数据的转换。下面是这些库之间常见的转换方法。

1. PIL和numpy之间的转换

import numpy as np
from PIL import Imageimage_path = "./"
pil_image = Image.open(image_path)
numpy_image = np.array(pil_image)
#从PIL转换到numpy
pil_image = Image.fromarray(numpy_image)
#从numpy转换到PIL

2. cv2和numpy之间的转换

import cv2
import numpy as npimage_path = "./"
opencv_image = cv2.imread(image_path)
numpy_image = np.array(opencv_image)
#从cv2到numpy的转换
opencv_image = cv2.cvtColor(numpy_image, cv2.COLOR_RGB2BGR)  
# 从numpy到cv2的转换。
#如果是RGB图像,要进行颜色空间转换

3.numpy和pytorch之间的转换

import torch
import numpy as npnumpy_image = np.array()  
# NumPy数组
torch_image = torch.from_numpy(numpy_image)
#从numpy到torch的转换
numpy_image = torch_image.numpy()
#从torch到numpy的转换

4.PIL和torch之间的转换

import torch
from PIL import Image
import torchvision.transforms as transformsimage_path = "./"
pil_image = Image.open(image_path)# 定义转换操作
transform = transforms.ToTensor()# 应用转换操作
torch_image = transform(pil_image)
#从PIL到torch的转换
pil_image = transforms.ToPILImage()(torch_image)
#从torch到PIL的转换

5.cv2和pytorch之间的转换

import torch
import cv2
import numpy as np
import torchvision.transforms as transformsimage_path = "./"
cv2_image = cv2.imread(image_path)
cv2_image = cv2.cvtColor(cv2_image, cv2.COLOR_BGR2RGB)  
# 转换为RGB格式# 定义转换操作
transform = transforms.Compose([transforms.ToTensor(),
])torch_image = transform(cv2_image)
# 将NumPy数组转换为PyTorch张量
cv2_image = cv2.cvtColor(torch_image, cv2.COLOR_RGB2BGR)
# 将torch转换为cv2

6.PIL和cv2之间的转换

import cv2
from PIL import Imageimage_path ="./"
pil_image = Image.open(image_path)# PIL图像转换为cv2图像
cv2_image = cv2.cvtColor(np.array(pil_image), cv2.COLOR_RGB2BGR)
# cv2图像转换为PIL图像
pil_image = Image.fromarray(cv2.cvtColor(cv2_image, cv2.COLOR_BGR2RGB))

http://www.lryc.cn/news/287073.html

相关文章:

  • Java面试题50道
  • 电脑怎么剪辑视频?这些软件不可错过
  • HBase学习七:Compaction
  • MySQL定期整理磁盘碎片
  • 【centos7安装docker】
  • 四、Flask学习之JavaScript
  • IO 专题
  • MySql索引事务讲解和(经典面试题)
  • 《微信小程序开发从入门到实战》学习九十一
  • 【立创EDA-PCB设计基础】6.布线铺铜实战及细节详解
  • Node.JS CreateWriteStream(大容量写入文件流优化)
  • 安卓开发之自动缩放布局
  • DDD系列 - 第9讲 实体、值对象
  • 5分钟做自己的微信红包封面
  • pytorch中BCELoss 和 binary_cross_entropy_with_logits之间的区别
  • 无刷电机学习-方波电调 程序篇1(AM32)
  • 如何自己制作一个属于自己的小程序?
  • HTML 入门手册(二)
  • 零基础学Python(5)— 基本数据类型
  • centos7安装Redis7.2.4
  • certificate-transparency-go用例
  • 前端上传大文件使用分片上传
  • Kotlin 解构声明
  • 基于若依的ruoyi-nbcio流程管理系统一种简单的动态表单模拟测试实现(四)
  • 遗传算法原理详细讲解(算法+Python源码)
  • Linux文本处理指令汇总
  • Prompt Engineering
  • Ansible剧本playbooks
  • Excel·VBA时间范围筛选及批量删除整行
  • Map转成String,String 转换成Map