【Python】使用Python处理RAW格式图片,并根据实际情况完成分组打包发送

背景

出游之后,朋友交换的照片格式大多是RAW格式,一些人想要JPG格式,但是百度云盘非会员的下载速度惨不忍睹,所以我想着通过微信群直接传(这个在事后也被证实不能完全解决问题,微信限制了每天传递文件的上限)。
出游之后,交换的照片格式大多是RAW格式,一些朋友想要JPG,因为百度云盘非会员的下载速度惨不忍睹,且受限于微信单文件100M,所以写了个Python脚本直接转换图片并进行打包发送。
下面的这个是在此过程中写的Python脚本,完成RAW格式图片的转换,并根据文件大小创建文件夹(微信的单文件大小为100M)进行打包发送。

库依赖

安装方法
Python3
rawpy pip install rawpy
imageio pip install imageio

rawpy的postprocess函数有很多方法,在文末附录中已给出,大家可以根据实际需要进行使用。

实际代码

import rawpy
import glob
import os
import imageio
import shutil
import zipfile


def raw2jpg(raw_file_name, dst="Temp_JPG", _suffix=".NEF"):
    """
    :param raw_file_name:
    :param dst: 存储目录
    :param _suffix: 文件后缀
    :return:
    """
    with rawpy.imread(raw_file_name) as raw:
        im = raw.postprocess(
            use_camera_wb=True,  # 是否使用拍摄时的白平衡值
            use_auto_wb=False,
            # half_size=True,  # 是否输出一半大小的图像,通过将每个2x2块减少到一个像素而不是进行插值来
            exp_shift=3  # 修改后光线会下降,所以需要手动提亮,线性比例的曝光偏移。可用范围从0.25(变暗2级)到8.0(变浅3级)。
        )
        imageio.imsave(dst + raw_file_name.strip(_suffix) + ".jpg", im)  # 因为glob函数返回的是一个相对路径,所以不需要使用os.path


def files2zip(files, _extra=".jpg", dst_zip_size=30):
    """
    :param files: 文件夹信息
    :param _extra: 后缀
    :param dst_zip_size:  目标文件大小,实际转换时会 * 0.8 防止大小溢出
    :return:
    """
    init_folder_num = 0
    folder_size = len(files)
    avg_file_size = sum([os.path.getsize(i) / float(1024 * 1024) for i in files]) / folder_size  # 计算转换后文件平均大小
    _split_num = int(dst_zip_size * 0.8 // avg_file_size)  # 计算切割的文件个数, 因为是平均值,* 0.8 防止溢出

    # 创建文件夹
    for i in range(folder_size // _split_num + 1):
        os.makedirs(f"Dst/part{i}", int("755", 8))  # 十进制转变为八进制

    # 移动文件
    for i in range(folder_size):
        if (folder_size - i) % _split_num:
            # shutil.move(JPG[i], f"Dst/part{init_folder_num}")
            shutil.copy(files[i], f"Dst/part{init_folder_num}")
            continue
        init_folder_num += 1
        # shutil.move(JPG[i], f"Dst/part{init_folder_num}")
        shutil.copy(files[i], f"Dst/part{init_folder_num}")

    # zip 打包
    for i in range(folder_size // _split_num + 1):
        folder_name = f"Dst/part{i}"
        z = zipfile.ZipFile(f"{folder_name}.zip", 'w')
        files = glob.glob(os.path.join(f"{folder_name}", f"*{_extra}"))
        for _file in files:
            z.write(_file)


src_suffix = ".NEF"
dst_suffix = ".jpg"
raw_files = glob.glob(f"NEF/*{src_suffix}")

print("正在转换中,请耐心等待....")
for num, raw_file in enumerate(raw_files):
    if num % 5 == 0: print(f"已转换{num}张照片...")
    raw2jpg(raw_file, _suffix=src_suffix)

print("转换完成!")

JPGs = glob.glob(f"Temp_JPG/*{dst_suffix}")  # 获取转换后的文件信息

files2zip(JPGs)
print("所有数据保留在Dst目录,请前往查看!")
print("Done!")

附录

参数 含义
demosaic_algorithm(rawpy.DemosaicAlgorithm) 默认为AHD
half_size(bool) 通过将每个2x2块减少到一个像素而不是进行插值来输出一半大小的图像
four_color_rgb(bool) 是否对两个绿色通道使用单独的插值
dcb_iterations(int) DCB校正通过的次数,需要DCB去马赛克算法
dcb_enhance(bool) 具有增强的插值颜色的DCB插值
fbdd_noise_reduction(rawpy.FBDDNoiseReductionMode) 在去马赛克之前控制FBDD降噪
noise_thr(float) 小波去噪的阈值(默认禁用)
mean_filter_passes(int) 去马赛克后减少颜色伪像的中间滤波器通过次数
use_camera_wb(bool) 是否使用拍摄时的白平衡值
use_auto_wb(bool) 是否尝试自动计算白平衡
user_wb(list) 长度4的列表,每种颜色都有白平衡倍增器
output_color(rawpy.ColorSpace) 输出颜色空间
output_bps(int) 8或16
user_flip(int) 0 =无,3 = 180、5 = 90CCW,6 = 90CW,默认为使用RAW图像的图像方向(如果可用)
user_black(int) 自定义黑电平
user_sat(int) 饱和度调整
no_auto_scale(bool) 是否禁用像素值缩放
no_auto_bright(bool) 是否禁用自动增加亮度
auto_bright_thr(float) 使用自动亮度增加时裁剪的像素比率(请参见no_auto_bright)。默认值为0.01(1%)。
Adjust_maximum_thr(float) 请参阅libraw文档
明亮(浮动) 亮度缩放
highlight_mode(rawpy.HighlightMode int)
exp_shift(float) 线性比例的曝光偏移。可用范围从0.25(变暗2级)到8.0(变浅3级)。
exp_preserve_highlights(float) 使用exp_shift增亮图像时保留高光。从0.0到1.0(完全保留)。
伽玛(元组) 对(幂,斜率),默认为(2.222,4.5)。BT.709
chromatic_aberration(元组) 对(red_scale,blue_scale),默认是(1,1),通过缩放红色和蓝色通道校正色像差
bad_pixels_path(str) dcraw坏像素文件的路径。每个坏像素将使用相邻像素的平均值进行校正。参见rawpy.enhance模块以了解其他修复算法,例如使用中位数。

参考文档:
https://github.com/letmaik/rawpy

posted @ 2019-12-09 10:41  _BingoHe  阅读(3699)  评论(1编辑  收藏  举报