快捷導(dǎo)航

Python實(shí)現(xiàn)批量設(shè)置圖片背景為透明

更新時間：2024年11月11日 09:06:48 作者：PythonFun

我們?nèi)粘Ｉ钪兄谱鱌PT等教學(xué)資源時,需要批量去除圖片背景,就可以使用 Python 的 rembg 庫,下面我們就來看看如何操作rembg實(shí)現(xiàn)批量設(shè)置圖片背景為透明吧

一、安裝rembg庫

不過，使用前需要在cmd下或者pycharm, thonny等pyhon的idle調(diào)試工具中安裝一下這個rembg庫，安裝代碼：

pip install rembg

二、下載u2net.onnx機(jī)器學(xué)習(xí)模型

安裝完成rembg模塊后，第一次運(yùn)行程序，調(diào)用這個模塊，還需要安裝機(jī)器學(xué)習(xí)模型：u2net.onnx，大小為176M，程序會自動下載，全程不需要干涉。

下載機(jī)器學(xué)習(xí)模型

這個模型文件 (u2net.onnx) 是經(jīng)過訓(xùn)練的深度學(xué)習(xí)模型，包含了用于識別并分離前景和背景的算法權(quán)重。rembg 庫使用 U2-Net 來執(zhí)行背景去除的任務(wù)。第一次使用 rembg 時，庫會自動下載這個模型文件，以確保能夠正確運(yùn)行背景去除功能。

具體來說：模型文件 u2net.onnx：U2-Net 模型是一個開源的圖像分割模型，專門用于前景-背景分離。rembg 使用這個模型的 .onnx 文件格式，方便兼容各種框架。

由于rembg 沒有默認(rèn)包含該模型文件，因此在第一次運(yùn)行時需要下載并存儲在本地（C:\Users\用戶名\.u2net\u2net.onnx），下載完成后，模型文件會保存在你用戶目錄的 .u2net 文件夾中，后續(xù)運(yùn)行 rembg 就不需要重新下載。

三、編寫清除圖片背景代碼

下載完成后，程序會自動運(yùn)行清除背景的程序，如下面的程序，批量把當(dāng)前目錄下的圖片文件，移除背景，設(shè)置透明背景后存在output文件夾下面，全程操作非常的快。

import os
from rembg import remove
from PIL import Image
 
# 定義輸入和輸出文件夾
input_folder = '.'  # 當(dāng)前目錄
output_folder = './output/'  # 輸出文件夾
 
# 創(chuàng)建輸出文件夾（如果不存在）
# if not os.path.exists(output_folder):
os.makedirs(output_folder,exist_ok=True)
 
# 遍歷當(dāng)前目錄的所有文件
for filename in os.listdir(input_folder):
    # 檢查是否為圖像文件（可以添加更多格式）
    if filename.lower().endswith(('.png', '.jpg', '.jpeg')):
        input_path = os.path.join(input_folder, filename)
        output_path = os.path.join(output_folder, filename.split('.')[0] + '_no_bg.png')
        
        # 打開圖像并去除背景
        with open(input_path, 'rb') as input_file:
            input_image = input_file.read()
            output_image = remove(input_image)
        
        # 將結(jié)果保存為 PNG 格式
        with open(output_path, 'wb') as output_file:
            output_file.write(output_image)
 
print("圖片背景已去除并保存到 'output' 文件夾。")

input_folder 指定輸入文件夾，這里設(shè)為當(dāng)前目錄 ./。

output_folder 是輸出文件夾（去除背景后的圖片保存位置），如果不存在，則會自動創(chuàng)建。

腳本遍歷當(dāng)前目錄中的所有圖像文件（.png, .jpg, .jpeg），去除背景后以透明背景 PNG 格式保存到 output 文件夾中。