Spaces:
Running
Running
[HANDCRAFT] Commit
Browse files- app.py +2 -2
- component/bbox.py +1 -1
app.py
CHANGED
@@ -70,8 +70,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as interface:
|
|
70 |
with gr.Column(scale=1):
|
71 |
output_bbox_result = gr.Textbox(label="🏷️ Number of Hands & Classification with Confidence")
|
72 |
include_standard = gr.Checkbox(
|
73 |
-
label="🤲 Include
|
74 |
-
value=
|
75 |
)
|
76 |
expand_ratio = gr.Slider(
|
77 |
minimum=0.5,
|
|
|
70 |
with gr.Column(scale=1):
|
71 |
output_bbox_result = gr.Textbox(label="🏷️ Number of Hands & Classification with Confidence")
|
72 |
include_standard = gr.Checkbox(
|
73 |
+
label="🤲 Include All Hands",
|
74 |
+
value=True
|
75 |
)
|
76 |
expand_ratio = gr.Slider(
|
77 |
minimum=0.5,
|
component/bbox.py
CHANGED
@@ -3,7 +3,7 @@ import cv2
|
|
3 |
import numpy as np
|
4 |
|
5 |
|
6 |
-
hand_dict = {0: '
|
7 |
|
8 |
def bbox(image, include_standard, mask_expand_ratio):
|
9 |
model = YOLO('model/yolo.pt')
|
|
|
3 |
import numpy as np
|
4 |
|
5 |
|
6 |
+
hand_dict = {0: '0', 1: '1'}
|
7 |
|
8 |
def bbox(image, include_standard, mask_expand_ratio):
|
9 |
model = YOLO('model/yolo.pt')
|