Spaces:
Runtime error
Runtime error
| 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() |