7.0 KiB
関連プロジェクト:
- SwanLab:トレーニング画像マットモデル全体を分析および監視するために使用され、研究室の学生と協力して交流し、トレーニング効率を大幅に向上させました。
🤩 プロジェクトの更新
-
2024.9.2: 写真の KB サイズを調整してを更新
-
2023.12.1: **API デプロイメント(fastapi ベース)**を更新
-
2023.6.20: プリセットサイズメニュー
-
2023.6.19: レイアウト写真を更新
-
2023.6.13: センターグラデーションカラーを更新
-
2023.6.11: 上下グラデーションカラーを更新
-
2023.6.8: カスタムサイズを更新
-
2023.6.4: カスタム背景色、顔検出バグ通知を更新
-
2023.5.10: サイズを変更せずに背景を変更を更新
概要
🚀 私たちの仕事に興味を持っていただきありがとうございます。画像処理分野での他の成果もご覧いただけます。お気軽に zeyi.lin@swanhub.co までご連絡ください。
HivisionIDPhoto は、実用的な証明写真のインテリジェント作成アルゴリズムを開発することを目的としています。
このアルゴリズムは、完全なモデルワークフローを使用して、さまざまなユーザーの写真シナリオを認識し、画像をセグメント化し、証明写真を生成します。
HivisionIDPhoto は以下のことができます:
- 軽量な画像セグメンテーション (高速推論には CPU のみが必要です。)
- 異なるサイズ仕様に基づいて標準的な証明写真や 6 インチのレイアウト写真を生成
- 美顔機能(待機中)
- インテリジェントなフォーマルウェアの交換(待機中)
HivisionIDPhoto が役に立つ場合は、このリポジトリにスターを付けるか、友人に推薦して、緊急の証明写真作成の問題を解決してください!
🔧 環境のインストールと依存関係
- Python >= 3.7(プロジェクトの主なテストは Python 3.10 で行われています)
- onnxruntime
- OpenCV
- オプション:Linux, Windows, MacOS
1. プロジェクトをクローン
git clone https://github.com/Zeyi-Lin/HivisionIDPhotos.git
cd HivisionIDPhotos
2. 依存パッケージをインストール
pip install -r requirements.txt
3. 重みファイルをダウンロード
Releaseから重みファイルhivision_modnet.onnxをダウンロードし、ルートディレクトリに保存します。
🚀 Gradio デモの実行
python app.py
プログラムを実行すると、ローカルの Web ページが生成され、そのページで証明写真の操作と対話を完了できます。
⚡️ API サービスのデプロイ
python deploy_api.py
API サービスのリクエスト(Python)
Python を使用してサービスにリクエストを送信します:
証明写真の作成(1 枚の写真を入力し、1 枚の標準証明写真と 1 枚の高解像度証明写真の 4 チャンネル透明 png を取得):
python requests_api.py -u http://127.0.0.1:8080 -i images/test.jpg -o ./idphoto.png -s '(413,295)'
背景色を追加(1 枚の 4 チャンネル透明 png を入力し、背景色が追加された画像を取得):
python requests_api.py -u http://127.0.0.1:8080 -t add_background -i ./idphoto.png -o ./idhoto_ab.jpg -c '(0,0,0)' -k 30
6 インチのレイアウト写真を取得(3 チャンネルの写真を入力し、6 インチのレイアウト写真を取得):
python requests_api.py -u http://127.0.0.1:8080 -t generate_layout_photos -i ./idhoto_ab.jpg -o ./idhoto_layout.jpg -s '(413,295)' -k 200
#🐳 Docker デプロイメント
1. イメージの取得またはビルド
イメージの取得:
このイメージは ARM アーキテクチャマシン(例:Mac M1)で構築されており、x86 アーキテクチャマシンを使用する場合は Dockerfile を使用してください。
docker pull linzeyi/hivision_idphotos:v1
Dockrfile によるイメージの構築:
hivision_modnet.onnxというモデルの重みファイルがルートディレクトリに配置されていることを確認した後、次のコマンドを実行します。
docker build -t hivision_idphotos .
2. Gradio Demo の実行
画像パッケージングが完了したら、次のコマンドを実行して 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
📖 参考プロジェクト
- MTCNN: https://github.com/ipazc/mtcnn
- ModNet: https://github.com/ZHKKKe/MODNet
💻 開発のヒント
1. デフォルトサイズを変更する方法
size_list_CN.csv を変更してから、app.py を再実行します。最初の列はサイズ名であり、2 番目の列は高さ、3 番目の列は幅です。
📧 お問い合わせ
ご質問がある場合は、zeyi.lin@swanhub.co までメールでお問い合わせください。
貢献者
Zeyi-Lin、SAKURA-CAT、Feudalman、swpfY、Kaikaikaifang、ShaohonChen、KashiwaByte

