基于Python的OpenCV库实现图像的安全读取与保存指南

def image_read(self)->np.ndarray:
    """安全读取中文路径的图像，支持单通道、三通道(BGR)、四通道(BGRA)"""
    # 二进制读取 + OpenCV解码
    source_path = Path(self.source_image_path)
    raw_data = np.fromfile(source_path, dtype=np.uint8)
    return cv2.imdecode(raw_data, cv2.IMREAD_UNCHANGED)  # 保留原始通道信息

def image_write(self, image:np.ndarray, file_extension:str ='.tif')->bool:
    # 根据不同格式设置优化参数
    if file_extension in ['.tif', '.tiff']:
        encode_params = [cv2.IMWRITE_TIFF_COMPRESSION, 1]  # 无压缩
    elif file_extension in ['.png']:
        encode_params = [cv2.IMWRITE_PNG_COMPRESSION, 0]   # 无压缩
    # ... 其他格式处理

# 单个图像文件的读取和保存
source_path = "输入图像路径.tif"
target_path = "输出图像路径.png"

image_io = ImageIO(source_path, target_path)
image = image_io.image_read()  # 读取图像
success = image_io.image_write(image, ".png")  # 保存为PNG格式

source_folder = Path("源文件夹")
target_folder = Path("目标文件夹")
file_io = FileIO(source_folder, target_folder)

# 获取所有图像文件路径
image_paths = file_io.filter_by_strings(file_io.list_file_paths(True), [".tif", ".jpg"], True)

for image_path in image_paths:
    source_image_path = Path(image_path)
    target_image_path = target_folder / source_image_path.name
    image_io = ImageIO(source_image_path, target_image_path)
    image = image_io.image_read()
    image_io.image_write(image, ".tif")  # 保持原格式或转换为指定格式

from pathlib import Path
import cv2
import numpy as np
from submodule.file_io import FileIO
#ImageIO类只涉及到图像数据的读取和储存，不涉及到图像数据的修改
class ImageIO:

    def __init__(self, source_image_path: None | str | Path, target_image_path: None | str | Path) -> None:
        """
        初始化 FileIO 实例

        Args:
            source_image_path: 读取图像路径。如果提供，会检查其是否存在。
            target_image_path: 保存图像路径。保存时必须提供
        """
        # 处理源文件夹：如果提供则转换为Path，并检查存在性（但不创建）
        self.source_image_path = Path(source_image_path) if source_image_path is not None else None
        if self.source_image_path is not None and not self.source_image_path.exists():
            # 根据你的需求决定：如果源文件夹必须存在，可以抛出异常
            # raise ValueError(f"源目录不存在: {self.source_folder}")
            # 或者仅记录警告，取决于你的业务逻辑
            print(f"警告: 源文件可能不存在: {self.source_image_path}")
        self.target_image_path = Path(target_image_path)

    def image_read(self)->np.ndarray:
        """安全读取中文路径的图像，支持单通道、三通道(BGR)、四通道(BGRA)"""
        # 二进制读取 + OpenCV解码
        source_path = Path(self.source_image_path)
        raw_data = np.fromfile(source_path, dtype=np.uint8)
        return cv2.imdecode(raw_data, cv2.IMREAD_UNCHANGED)  # 保留原始通道信息

    def image_write(self, image:np.ndarray, file_extension:str ='.tif')->bool:
        """安全保存图像（支持中文路径），支持多种格式的无损保存

        Args:
            image: 要保存的图像数据
            file_extension: 文件扩展名，支持 '.tif','.tiff', '.png', '.jpg', '.jpeg', '.webp' 等

        Returns:
            bool: 保存成功返回True，否则返回False
        """
        target_image_path= Path(self.target_image_path)


        # 根据文件扩展名设置不同的无损保存参数 [1,7,10](@ref)
        file_extension = file_extension.lower()

        if file_extension in ['.tif', '.tiff']:
            # TIFF格式：使用无压缩 [7](@ref)
            encode_params = [cv2.IMWRITE_TIFF_COMPRESSION, 1]
        elif file_extension in ['.png']:
            # PNG格式：压缩级别0（不压缩）[7](@ref)
            encode_params = [cv2.IMWRITE_PNG_COMPRESSION, 0]
        elif file_extension in ['.jpg', '.jpeg']:
            # JPEG格式：最高质量100（虽然JPEG固有有损，但100是最小损失）[1](@ref)
            encode_params = [cv2.IMWRITE_JPEG_QUALITY, 100]
        elif file_extension in ['.webp']:
            # WebP格式：最高质量100 [10](@ref)
            encode_params = [cv2.IMWRITE_WEBP_QUALITY, 100]
        else:
            # 其他格式不使用压缩参数
            encode_params = []

        # 使用imencode避免中文路径问题 [8](@ref)
        success, encoded_image = cv2.imencode(file_extension, image, params=encode_params)

        if success:
            encoded_image.tofile(target_image_path)
            return True
        else:
            print(f"图像保存失败，格式: {file_extension}")
            return False
if __name__ == "__main__":
    source_folder = Path(r"source")
    target_folder = Path(r"target")
    file_io = FileIO(source_folder, target_folder)#创建FileIO实例处理文件列表
    file_paths = file_io.list_file_paths(True) #获取所有文件路径
    image_paths = file_io.filter_by_strings(file_paths,[".tif"],True)#获取.tif文件的路径
    file_io.clear_folder(target_folder)# 清空目标文件夹（谨慎使用）
    # 处理每个图像
    for image_path in image_paths:
        source_image_path = Path(image_path) #目标图像路径
        target_image_path = target_folder / source_image_path.name #要保存的图像路径，可以修改名称来重命名
        image_io = ImageIO(source_image_path,target_image_path) #实例化ImageIO类
        image = image_io.image_read() #读取目标图像数据，BGR

        # 可在此处添加图像处理代码
        # 例如：调整大小、应用滤镜等

        image_io.image_write(image,file_extension=".tif") #保存图像数据至目标文件夹，可以设置指定格式
        print("成功读取并保存：", f"{source_image_path} -> {target_image_path}")

raw_data = np.fromfile(source_path, dtype=np.uint8)
image = cv2.imdecode(raw_data, cv2.IMREAD_UNCHANGED)

target_path.parent.mkdir(parents=True, exist_ok=True)

# 将BGR转换为RGB
image_rgb = cv2.cvtColor(image_bgr, cv2.COLOR_BGR2RGB)