rein0421 commited on
Commit
146e98d
·
verified ·
1 Parent(s): 2ba2f95

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +26 -0
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