Spaces:
Running
Running
File size: 1,754 Bytes
fb39199 aa7e786 fb39199 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
import os
import gradio as gr
from imgutils.generic import YOLOModel
_GLOBAL_CSS = """
.limit-height {
max-height: 55vh;
}
"""
if __name__ == '__main__':
with gr.Blocks(css=_GLOBAL_CSS) as demo:
with gr.Row():
with gr.Column():
gr.HTML('<h2 style="text-align: center;">Object Detections For Real Photos</h2>')
gr.Markdown('This is the online demo for detection functions of '
'[imgutils.detect](https://dghs-imgutils.deepghs.org/main/api_doc/detect/index.html). '
'You can try them yourselves with `pip install dghs-imgutils`.')
with gr.Row():
with gr.Tabs():
with gr.Tab('Face Detection'):
YOLOModel('deepghs/real_face_detection').make_ui(
default_model_name='face_detect_v0_s_yv11',
default_conf_threshold=0.25,
)
with gr.Tab('Face Detection (Real Only)'):
YOLOModel('deepghs/yolo-face').make_ui(
default_model_name='yolov10s-face',
)
with gr.Tab('Head Detection'):
YOLOModel('deepghs/real_head_detection').make_ui(
default_model_name='head_detect_v0_s_yv11',
default_conf_threshold=0.18,
)
# with gr.Tab('Person Detection'):
# PersonDetection().make_ui()
with gr.Tab('Generic'):
YOLOModel('deepghs/yolos').make_ui(
default_model_name='yolo11s',
)
demo.queue(os.cpu_count()).launch()
|