Spaces:
Running
Running
Update README.md
Browse files
README.md
CHANGED
@@ -6,5 +6,31 @@ colorTo: pink
|
|
6 |
sdk: docker
|
7 |
pinned: false
|
8 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
9 |
|
10 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
|
|
6 |
sdk: docker
|
7 |
pinned: false
|
8 |
---
|
9 |
+
処理の流れ
|
10 |
+
ユーザーが画像をアップロードします。
|
11 |
+
アップロードされた画像は、save_image関数でサーバーに保存されます。
|
12 |
+
リスクレベルと画像パスを基に、process_image関数が画像認識を行い、マスク画像を生成します。
|
13 |
+
マスク画像と元画像を用いて、以下のいずれかの関数で画像の不要部分を補正します:
|
14 |
+
inpaint_image_with_mask(OpenCVを使用)
|
15 |
+
inpaint_image_with_mask1(Simple Lamaを使用)
|
16 |
+
create_mask_and_inpaint_stable_diffusion(Stable Diffusionを使用)
|
17 |
+
補正された画像をユーザーに返します。
|
18 |
+
各関数の説明
|
19 |
+
quantize_model: モデルを量子化し軽量化する関数。CPU実行時に使用します。
|
20 |
+
decide_to_object: リスクレベルに基づき、消去するオブジェクトを決定する関数。
|
21 |
+
create_mask: オブジェクトの位置情報を基にマスク画像を作成する関数。
|
22 |
+
special_process_image: 指定された領域以外を画像認識し、マスク画像を生成する関数。
|
23 |
+
process_image: 画像認識を行い、マスク画像を作成する関数。内部でdecide_to_objectおよびcreate_maskを使用。
|
24 |
+
inpaint_image_with_mask: OpenCVのインペイント機能を用い、マスク画像を基に元画像を補正する関数。
|
25 |
+
inpaint_image_with_mask1: Simple Lamaを用い、マスク画像を基に元画像を補正する関数。
|
26 |
+
create_mask_and_inpaint_stable_diffusion: Stable Diffusionを用い、マスク画像を基に元画像を補正する関数。
|
27 |
+
save_image: アップロードされた画像をサーバーに保存する関数。
|
28 |
+
copy_image_to_folder: DeepFillv2で処理するために画像を指定のフォルダにコピーする関数。
|
29 |
+
run_python_script: DeepFillv2の実行用Pythonスクリプトを実行する関数。
|
30 |
+
process_image_vec: 画像をベクトル化する関数。
|
31 |
+
classify_new_image: 新しい画像を既存のクラスタに分類する関数。
|
32 |
+
run_fastapi・run_ngrok: FastAPIとngrokを並行実行するための関数。
|
33 |
+
save_sums_to_json: クラスタリング結果をJSONファイルに保存する関数。
|
34 |
+
|
35 |
|
36 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|