zhenyueqin commited on
Commit
acf80b4
·
verified ·
1 Parent(s): 6d1116c

[HANDCRAFT] Commit

Browse files
Files changed (2) hide show
  1. app.py +2 -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 Standard Hands",
74
- value=False
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: 'Standard', 1: 'Non-Standard'}
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')