File size: 1,190 Bytes
419a809
32e64e6
 
60dd03b
 
419a809
60dd03b
419a809
 
 
 
 
 
60dd03b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
32e64e6
 
 
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
---
title: Object Detection - YOLOV3
emoji: πŸš€
colorFrom: green
colorTo: purple
sdk: gradio
sdk_version: 3.40.1
app_file: app.py
pinned: false
license: mit
---

Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

Demonstration of Object Detection using YOLOv3. This model is trained using Pytorch Lightning from scratch on PASCAL VOC dataset with 20 classes.
The input image resolution used for the training was 416 x 416. 

The app also shows the saliency maps of the input images generated using EigenCAM explainability method. 

The model supports following classes:
  
   * aeroplane ✈️
   * bicycle 🚲
   * bird 🐦
   * boat πŸ›₯️
   * bottle 🍾
   * bus 🚌
   * car πŸš—
   * cat 🐈
   * chair πŸͺ‘
   * cow πŸ„
   * diningtable 
   * dog πŸ•
   * horse 🐎
   * motorbike 🏍️
   * person πŸ‘±
   * pottedplant
   * sheep πŸ‘
   * sofa πŸ›‹οΈ
   * train 🚊
   * tvmonitor πŸ“Ί

### Steps to Use:
* Upload an image and click submit to generate the detections, the corresponding bounding boxes, and the saliency maps.
* Control the degree of detection threshold and the IOU threshold to filter the predictions