随手笔记-python-opencv 读取图像的顺序 与pytorch处理图像的顺序

python 复制代码
import cv2

# 读取图像
image_path = 'path/to/your/image.jpg'  # 替换为你的图像路径
image = cv2.imread(image_path)

# 检查图像是否成功读取
if image is None:
    print("Error: Unable to load image.")
else:
    print("Image loaded successfully.")

1、OpenCV 使用 BGR 格式而不是 RGB 格式。

python 复制代码
import matplotlib.pyplot as plt

# 将 BGR 图像转换为 RGB
rgb_image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)

# 使用 Matplotlib 显示图像
plt.imshow(rgb_image)
plt.title('Image')
plt.axis('off')  # 关闭坐标轴
plt.show()

在 PyTorch 中处理图像时,图像的维度顺序通常与 NumPy 或 OpenCV 中的顺序有所不同。PyTorch 的图像处理顺序通常遵循 [C, H, W] 的维度顺序,即:

  • C:通道数(Channels),例如 RGB 图像有 3 个通道,灰度图像有 1 个通道。

  • H:图像高度(Height)。

  • W:图像宽度(Width)。

2、在使用 PyTorch 处理图像时,通常需要将图像从 [H, W, C] 转换为 [C, H, W] 的格式。以下是常见的处理流程:

python 复制代码
import torch
import numpy as np
from PIL import Image

# 加载图像
image_path = "path/to/image.jpg"
image_pil = Image.open(image_path)

# 转换为 NumPy 数组
image_np = np.array(image_pil)

# 调整维度顺序:从 [H, W, C] 转换为 [C, H, W]
image_np = image_np.transpose((2, 0, 1))

# 转换为 PyTorch 张量
image_tensor = torch.from_numpy(image_np).float()

# 归一化到 [0, 1]
image_tensor = image_tensor / 255.0

print("Image tensor shape:", image_tensor.shape)
print("Image tensor dtype:", image_tensor.dtype)

3、 PyTorch 提供了 torchvision.transforms 模块,可以更方便地处理图像。以下是一个使用 torchvision.transforms 的示例:

python 复制代码
import torch
from torchvision import transforms
from PIL import Image

# 定义图像预处理流程
transform = transforms.Compose([
    transforms.ToTensor(),  # 转换为 [C, H, W] 并归一化到 [0, 1]
    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])  # 标准化
])

# 加载图像
image_path = "path/to/image.jpg"
image_pil = Image.open(image_path)

# 应用预处理
image_tensor = transform(image_pil)

print("Image tensor shape:", image_tensor.shape)
print("Image tensor dtype:", image_tensor.dtype)
相关推荐
逼子格6 分钟前
逻辑门电路Multisim电路仿真汇总——硬件工程师笔记
笔记·硬件工程师·multisim·电路仿真·逻辑门·硬件工程师学习·电路图
@Hwang13 分钟前
【ESP32-IDF笔记】09-UART配置和使用
笔记·esp32·uart·esp32s3·esp32-idf
jndingxin23 分钟前
OpenCV 人脸分析----人脸识别的一个经典类cv::face::EigenFaceRecognizer
人工智能·opencv·计算机视觉
用什么都重名1 小时前
MinerU:高效智能PDF文档解析工具完全指南
人工智能·python·pdf·mineru·makedown
霖001 小时前
C++学习笔记三
运维·开发语言·c++·笔记·学习·fpga开发
倔强青铜三1 小时前
苦练Python第4天:Python变量与数据类型入门
前端·后端·python
这我可不懂1 小时前
Python 项目快速部署到 Linux 服务器基础教程
linux·服务器·python
倔强青铜三1 小时前
苦练Python第3天:Hello, World! + input()
前端·后端·python
小白学大数据1 小时前
Python爬取闲鱼价格趋势并可视化分析
开发语言·python
七月初七淮水竹亭~2 小时前
Pycharm 报错 Environment location directory is not empty 如何解决
ide·python·pycharm