Upload 5 files

README.md

@@ -1,13 +1,12 @@
 ---
-title: Yolov3 Object Detection
-emoji: 👁
-colorFrom: pink
-colorTo: green
+title: Yolo Object Detection
+emoji: 🏢
+colorFrom: yellow
+colorTo: gray
 sdk: gradio
-sdk_version: 3.41.1
+sdk_version: 3.40.1
 app_file: app.py
 pinned: false
-license: apache-2.0
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,29 @@
+import numpy as np
+import gradio as gr
+from src.detect import predict
+
+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()

model_ass_13_up.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a91aee65bdd1436341e1bba38729a39907e19ff520729e0ae8608984386e10d4
+size 740110073

requirements.txt

@@ -0,0 +1,13 @@
+torch
+torchvision
+grad-cam
+opencv-python
+albumentations
+pytorch_lightning
+torch-lr-finder
+pillow
+numpy
+pytorch-gradcam
+gradio
+seaborn
+

weights.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d1c64c5393b427f5f43a50736a242e95546f6dd646bf217f8c718940639c9072
+size 134