一鍵生成！輕量級 AI 證件照製作工具！

Java陈序员發表於2024-09-17
AI

大家好，我是 Java陳序員。

在日常生活中，我們需要各式各樣的證件照。有時候需要不同的尺寸，一寸、兩寸、小二寸...而有的時候需要不同的背景，白底、藍底、紅底...

拍攝不同格式的證件照，需要我們往返於攝像館，十分麻煩，又需要支出一定的花費！

今天，給大家介紹一款免費的 AI 證件照製作工具，使用 AI 幫助我們一鍵生成不同尺寸、背景的證件照！

關注微信公眾號：【Java陳序員】，獲取開源專案分享、AI副業分享、超200本經典計算機電子書籍等。

專案介紹

HivisionIDPhotos —— 一個輕量級的 AI 證件照製作工具，利用一套完善的模型工作流程，實現對多種使用者拍照場景的識別、摳圖與證件照生成。

功能特色：

  • 輕量級摳圖：僅需 CPU 即可快速推理
  • 功能齊全：根據不同尺寸規格生成不同的標準證件照、六寸排版照
  • 快速高效：能夠在幾秒鐘內處理影像，生成符合標準的證件照
  • 跨平臺支援：支援多種作業系統，無論是 Windows、macOS 還是 Linux 使用者都能輕鬆使用
  • 易於整合：提供清晰的 API 文件和示例程式碼，方便開發者快速整合到自己的專案中

專案體驗

HivisionIDPhotos 提供了一個簡單易用的 Web 介面供使用，我們只需要上傳影像，選定好預設尺寸、背景色、渲染方式、設定 KB 大小等就能快速生成標準照、高畫質照和六寸排版照。

在預設尺寸中，工具幫我們設定好了一些常用證件照的尺寸，如教師資格證、英語四六級考試、電子駕駛證等，我們直接選定使用。當然也可以自定義尺寸！

背景色中，除了提供常用的藍底、白底、紅底外，還支援自定義顏色。

自定義顏色需要輸入對應的紅（R）、綠（G）、藍（B）值。

可能有時候因為一些官方要求，生成的證件照有大小要求，HivisionIDPhotos 也支援我們在生成證件照的時候，設定大小。

下面是一些生成證件照的效果圖：

快速上手

環境安裝與依賴

環境依賴：

  • Python >= 3.7
  • onnxruntime
  • OpenCV
  • Option: Linux, Windows, MacOS

1、拉取程式碼

git clone https://github.com/Zeyi-Lin/HivisionIDPhotos.git

2、安裝依賴環境

cd  HivisionIDPhotos
pip install -r requirements.txt
pip install -r requirements-app.txt

3、下載權重檔案 hivision_modnet.onnx (24.7MB)，並存到專案的 hivision/creator/weights 目錄下

下載地址：https://github.com/Zeyi-Lin/HivisionIDPhotos/releases/tag/pretrained-model

執行 Gradio Demo

python app.py

執行程式將生成一個本地 Web 頁面，在頁面中可完成證件照的操作與互動。

啟動成功後，瀏覽器訪問：http://localhost:7860/

執行 API 服務

python deploy_api.py

API 服務提供了以下 3 個 API:

1、生成證件照（底透明）：傳送一張 RGB 影像，輸出一張標準證件照和一張高畫質證件照

curl -X POST "http://127.0.0.1:8080/idphoto" \
-F "input_image=@demo/images/test.jpg" \
-F "height=413" \
-F "width=295"

2、新增背景色：送一張 RGBA 影像，根據 color 新增背景色，合成一張 JPG 影像

curl -X POST "http://127.0.0.1:8080/add_background" \
-F "input_image=@test.png" \
-F "color=638cce" \
-F "kb=200"

3、生成六寸排版照：傳送一張 RGB 影像（一般為新增背景色之後的證件照），根據 size 進行照片排布，然後生成一張六寸排版照

curl -X POST "http://127.0.0.1:8080/generate_layout_photos" \
-F "input_image=@test.jpg" \
-F "height=413" \
-F "width=295" \
-F "kb=200"

專案提供了 cURL、Python、Java、JavaScript 的請求示例，詳細的使用可參考文件：

https://github.com/Zeyi-Lin/HivisionIDPhotos/blob/master/docs/api_CN.md

Docker 部署

1、拉取映象

docker pull linzeyi/hivision_idphotos:v1

2、執行 Gradio Demo

docker run -p 7860:7860 hivision_idphotos

在你的本地訪問 http://127.0.0.1:7860 即可使用。

3、執行 API 後端服務

docker run -p 8080:8080 hivision_idphotos python3 deploy_api.py

HivisionIDPhotos 利用先進的 AI 技術，自動識別人臉並調整照片的背景、光線和尺寸，確保輸出的證件照滿足各種官方要求。大家快去安裝體驗吧~

專案地址：https://github.com/acmenlei/codecv

最後

推薦的開源專案已經收錄到 GitHub 專案，歡迎 Star：

https://github.com/chenyl8848/great-open-source-project

或者訪問網站，進行線上瀏覽：

https://chencoding.top:8090/#/

大家的點贊、收藏和評論都是對作者的支援，如文章對你有幫助還請點贊轉發支援下，謝謝！

相關文章