Hugging Face's logo

Spaces:
wgetdd
/
YoloV3-PASCAL-VOC
Runtime error

App Files Community
YoloV3-PASCAL-VOC
File size: 2,568 Bytes 
cda4e7a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
 
import numpy as np
import gradio as gr
from src.detect import predict

from src.config import PASCAL_CLASSES


# def inference(
#     org_img: np.ndarray,
#     iou_thresh: float, thresh: float,
#     show_cam: str,
#     transparency: float,
# ):
#     outputs = predict(org_img, iou_thresh, thresh, show_cam, transparency)
#     return outputs

# demo = gr.Interface(
#     predict,
#     inputs=[
#         gr.Image(label="Input Image"),
#         gr.Slider(0, 1, value=0.5, label="IOU Threshold"),
#         gr.Slider(0, 1, value=0.4, label="Threshold"),
#         gr.Checkbox(label="Show Grad Cam"),
#         gr.Slider(0, 1, value=0.5, label="Opacity of GradCAM"),
#     ],
#     outputs=[
#         gr.Gallery(rows=2, columns=1),
#     ],
#     title="YoloV3 on PASCAL VOC Dataset From Scratch",
#     examples=[
#         ["example_images/009922.jpg", 0.5, 0.4, True, 0.5],
#         ["example_images/009938.jpg", 0.6, 0.5, True, 0.5],
#         ["example_images/009948.jpg", 0.55, 0.45, True, 0.5],
#         ["example_images/009952.jpg", 0.5, 0.4, True, 0.5],
#         ["example_images/009953.jpg", 0.6, 0.7, True, 0.5],
#         ["example_images/009956.jpg", 0.5, 0.4, True, 0.5],
#         ["example_images/009957.jpg", 0.6, 0.5, True, 0.5],
#         ["example_images/009960.jpg", 0.55, 0.45, True, 0.5],
#         ["example_images/009961.jpg", 0.5, 0.4, True, 0.5],
#         ["example_images/009962.jpg", 0.6, 0.7, True, 0.5],
#     ],
# )
# demo.launch()


gr.Interface(
    predict,
    inputs=[
        gr.Image(label="Input Image"),
        gr.Slider(0, 1, value=0.5, label="IOU Threshold"),
        gr.Slider(0, 1, value=0.4, label="Threshold"),
        gr.Checkbox(label="Show Grad Cam"),
        gr.Slider(0, 1, value=0.5, label="Opacity of GradCAM"),
    ],
    outputs=gr.Gallery(rows=2, columns=1),
    title="YoloV3 on PASCAL VOC Dataset From Scratch (Slide for GradCam output)",
    examples=[
        ["example_images/009922.jpg", 0.5, 0.4, True, 0.5],
        ["example_images/009938.jpg", 0.6, 0.5, True, 0.5],
        ["example_images/009948.jpg", 0.55, 0.45, True, 0.5],
        ["example_images/009952.jpg", 0.5, 0.4, True, 0.5],
        ["example_images/009953.jpg", 0.6, 0.7, True, 0.5],
        ["example_images/009956.jpg", 0.5, 0.4, True, 0.5],
        ["example_images/009957.jpg", 0.6, 0.5, True, 0.5],
        ["example_images/009960.jpg", 0.55, 0.45, True, 0.5],
        ["example_images/009961.jpg", 0.5, 0.4, True, 0.5],
        ["example_images/009962.jpg", 0.6, 0.7, True, 0.5],
    ],
    layout="horizontal"
).launch()