Spaces:
Running
on
Zero
Running
on
Zero
first
Browse files
app.py
CHANGED
|
@@ -178,7 +178,7 @@ def inference_pipeline(
|
|
| 178 |
@spaces.GPU
|
| 179 |
def gpu_inference(
|
| 180 |
image: Image.Image,
|
| 181 |
-
score_thresh: float = 0.
|
| 182 |
selected_classes: List[str] = None,
|
| 183 |
):
|
| 184 |
"""Spaces ZeroGPU が検出できるようにデコレータ付きの推論関数を用意"""
|
|
@@ -216,13 +216,13 @@ with gr.Blocks(title="DEIMv2 Floorplan Symbol Detection") as demo:
|
|
| 216 |
score_thresh = gr.Slider(
|
| 217 |
minimum=0.0,
|
| 218 |
maximum=1.0,
|
| 219 |
-
value=0.
|
| 220 |
step=0.05,
|
| 221 |
label="スコア閾値",
|
| 222 |
)
|
| 223 |
selected_classes = gr.CheckboxGroup(
|
| 224 |
choices=class_names,
|
| 225 |
-
value=
|
| 226 |
label="検出するクラス",
|
| 227 |
info="選択したクラスの検出結果のみが表示されます",
|
| 228 |
)
|
|
|
|
| 178 |
@spaces.GPU
|
| 179 |
def gpu_inference(
|
| 180 |
image: Image.Image,
|
| 181 |
+
score_thresh: float = 0.9, # UIのデフォルト値と統一
|
| 182 |
selected_classes: List[str] = None,
|
| 183 |
):
|
| 184 |
"""Spaces ZeroGPU が検出できるようにデコレータ付きの推論関数を用意"""
|
|
|
|
| 216 |
score_thresh = gr.Slider(
|
| 217 |
minimum=0.0,
|
| 218 |
maximum=1.0,
|
| 219 |
+
value=0.9,
|
| 220 |
step=0.05,
|
| 221 |
label="スコア閾値",
|
| 222 |
)
|
| 223 |
selected_classes = gr.CheckboxGroup(
|
| 224 |
choices=class_names,
|
| 225 |
+
value=["door1"], # デフォルトでdoor1のみ選択
|
| 226 |
label="検出するクラス",
|
| 227 |
info="選択したクラスの検出結果のみが表示されます",
|
| 228 |
)
|