【原】教你用 Python 快速批量轉換 HEIC 文件

AirPython 2021-12-23

展開全文

1. 前言

大家好，我是安果,！

最近打算做一批日歷給親朋好友,，但是從 iPhone 上導出的照片格式是 HEIC 格式，而商家的在線制作網站不支持這種圖片格式

PS：HEIC 是蘋果采用的新的默認圖片格式,，它能在不損失圖片畫質的情況下,，減少圖片大小

有很多在線網站支持圖片批量轉換，但是安全隱私又沒法得到保證,；如果使用 PS 等軟件去一張張轉換,，浪費時間的同時效率太低

本篇文章將使用 Python 批量實現 HEIC 圖片文件的格式轉換

2. 準備

首先，我們安裝 pyheif 依賴包

Linux 和 Mac OS 可以通過下面鏈接選擇合適的方式進行安裝

https:///project/pyheif/

如果是 Windows,，我們只能下載 whl 依賴文件,，使用 pip 命令進行安裝

下載地址：

https://www.lfd./~gohlke/pythonlibs/#pyheif

注意：我們需要根據系統(tǒng)及 Python 版本選擇對應的文件進行安裝

# 比如，本機是win10+64位 + Python3.7
# 通過下面鏈接下載文件：pyheif?0.6.1?cp37?cp37m?win_amd64.whl
https://www.lfd./~gohlke/pythonlibs/#pyheif

# 然后進行虛擬環(huán)境
# 使用pip3命令安裝whl文件
pip3 install pyheif?0.6.1?cp37?cp37m?win_amd64.whl

然后,，安裝 PIL 依賴,，用于圖片處理

# 安裝依賴
pip3 install Pillow

3. 實戰(zhàn)

首先，遍歷源文件夾及子文件夾,，獲取所有 HEIC 格式（不區(qū)分大小寫）的圖片

import pathlib
import os

def get_all_heic_imgs():
    """
    獲取所有heic格式的圖片
    :return:
    """
    # heif_image_paths = glob.glob(r"{}/*.heic".format(source_path))
    # 滿足條件的文件列表
    filelist = []
    for root, dirnames, filenames in os.walk(source_path):
        for filename in filenames:
            # filename：文件名,、root：文件對應的目錄
            # 獲取文件后綴名
            file_end = pathlib.Path(filename).suffix
            # 文件名（不帶后綴）
            file_name = pathlib.Path(filename).name.split(".")[-2]
            if file_end in ['.heic', '.HEIC']:
                # 文件的完整目錄
                # file_path = os.path.join(root, filename)
                filelist.append({
                    "filename": file_name,
                    "filepath": os.path.join(root, filename)
                })
    return filelist

然后，遍歷文件列表,，使用 pyheif 讀取文件,，使用PIL 轉為二進制圖片，以 JPG 格式保存到目標文件夾下

import pyheif
from PIL import Image

# 讀取文件
img = pyheif.read(filepath)
img_bytes = Image.frombytes(mode=img.mode, size=img.size, data=img.data)


# 文件保存完整目錄
target_file_path = '{}/{}_{}.jpg'.format(target_path, filename, generate_random_num(6))

# 保存
img_bytes.save(target_file_path, format="jpeg")

由于圖片數目很多,，圖片讀取,、圖片保存都是耗時的 IO 操作，最后對程序進行改造,，利用多線程加快圖片轉換

另外,，圖片可能會存在文件名重名，最后保存的文件名追加了一個隨機的字符串

import threading

def generate_random_num(count):
    """
    產生一段隨機的字符串
    :param count:
    :return:
    """
    return ''.join(random.sample('abcdefghijklmnopqrstuvwxyz', count))
    
def convert_heic_to_jpg(file, semaphore):
    """
    heic格式轉jpg
    :param files:
    :return:
    """
    semaphore.acquire()
    
    ...
    #文件操作

    # 釋放
    semaphore.release()
    
if __name__ == '__main__':
    ...
    # 定義信號量,，并發(fā)處理文件IO
    semaphore = threading.BoundedSemaphore(20)
    for file in files:
        t = threading.Thread(target=convert_heic_to_jpg, args=(file, semaphore))
        t.start()