pleonard commited on
Commit
0ed78ff
β€’
1 Parent(s): 38d54a3

add color metadat

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -8,15 +8,17 @@ import time
8
 
9
  dbackends = [
10
  ['Haar Cascade (OpenCV)','opencv'],
11
- ['Single Shot MultiBox Detector (OpenCV)','ssd'],
12
  ['Histogram of Oriented Gradients (Dlib)','dlib'],
13
  ['RetinaFace','retinaface'],
14
  ['You Only Look Once v8','yolov8'],
15
- ['YuNet','yunet'],
16
  # ['Multi-task Cascade Convolutional Neural Network (TensorFlow) ','mtcnn'],
17
  ['Fast Multi-task Cascade Convolutional Neural Network (PyTorch)','fastmtcnn']
18
  ]
19
 
 
 
20
 
21
  with gr.Blocks() as demo:
22
 
@@ -56,7 +58,7 @@ with gr.Blocks() as demo:
56
  print(run_time)
57
  return(results,numberoffaces,run_time,)
58
 
59
- dbackendchoice = gr.Radio(choices=dbackends,label='Detector Backend:',value='retinaface')
60
  gr.Interface(
61
  allow_flagging = "never",
62
  fn=findFaces,
 
8
 
9
  dbackends = [
10
  ['Haar Cascade (OpenCV)','opencv'],
11
+ ['🌈 Single Shot MultiBox Detector (OpenCV)','ssd'],
12
  ['Histogram of Oriented Gradients (Dlib)','dlib'],
13
  ['RetinaFace','retinaface'],
14
  ['You Only Look Once v8','yolov8'],
15
+ ['🌈 YuNet','yunet'],
16
  # ['Multi-task Cascade Convolutional Neural Network (TensorFlow) ','mtcnn'],
17
  ['Fast Multi-task Cascade Convolutional Neural Network (PyTorch)','fastmtcnn']
18
  ]
19
 
20
+ dbackendinfo = 'Detectors with 🌈 require a color image.'
21
+
22
 
23
  with gr.Blocks() as demo:
24
 
 
58
  print(run_time)
59
  return(results,numberoffaces,run_time,)
60
 
61
+ dbackendchoice = gr.Radio(choices=dbackends,label='Detector Backend:',info=dbackendinfo,container=True,value='retinaface')
62
  gr.Interface(
63
  allow_flagging = "never",
64
  fn=findFaces,