Spaces:
Sleeping
Sleeping
Commit ·
5f7693c
1
Parent(s): 9fdde30
Update app.py
Browse files
app.py
CHANGED
|
@@ -43,6 +43,17 @@ import litmodelclass
|
|
| 43 |
|
| 44 |
|
| 45 |
# gradio
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 46 |
title = "Yolo3 trained on PASCAL_VOC with GradCAM"
|
| 47 |
description = "Gradio interface to infer on Yolo3 model, and get GradCAM results"
|
| 48 |
|
|
@@ -185,5 +196,13 @@ with gr.Blocks() as demo:
|
|
| 185 |
outputs=img_output,
|
| 186 |
fn=yolo3_inference, cache_examples=False)
|
| 187 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 188 |
if __name__ == "__main__":
|
| 189 |
demo.launch()
|
|
|
|
| 43 |
|
| 44 |
|
| 45 |
# gradio
|
| 46 |
+
|
| 47 |
+
model_stats = """
|
| 48 |
+
### YoloV3 Model Implementation & Training Details
|
| 49 |
+
Github Link: https://github.com/santule/ERA/tree/main/S13
|
| 50 |
+
#### Model Performance:
|
| 51 |
+
1. **Validation Loss: 6.05**
|
| 52 |
+
2. **Class accuracy: 82.4%**
|
| 53 |
+
3. **No obj accuracy: 98.05%**
|
| 54 |
+
4. **Obj accuracy: 72.3%**
|
| 55 |
+
"""
|
| 56 |
+
|
| 57 |
title = "Yolo3 trained on PASCAL_VOC with GradCAM"
|
| 58 |
description = "Gradio interface to infer on Yolo3 model, and get GradCAM results"
|
| 59 |
|
|
|
|
| 196 |
outputs=img_output,
|
| 197 |
fn=yolo3_inference, cache_examples=False)
|
| 198 |
|
| 199 |
+
|
| 200 |
+
with gr.Row():
|
| 201 |
+
with gr.Box():
|
| 202 |
+
with gr.Row():
|
| 203 |
+
with gr.Column():
|
| 204 |
+
with gr.Box():
|
| 205 |
+
gr.Markdown(model_stats)
|
| 206 |
+
|
| 207 |
if __name__ == "__main__":
|
| 208 |
demo.launch()
|