README.md

metadata

title: TSAI S13
emoji: 🔥
colorFrom: gray
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

The School of AI - ERA(Extensive & Reimagined AI Program) - Assignment 13

This folder consists of Assignment-13 from ERA course offered by - TSAI(The school of AI). Follow https://theschoolof.ai/ for more updates on TSAI

For more details on the assignment, refer to github link: https://github.com/ToletiSri/TSAI_ERA_Assignments/tree/main/S13

As part of the assignment, we have trained a YOLO-V3 model on PASCAL-VOC dataset, using pytorch lightning. PASCAL-VOC dataset is about 5GB of Data. Training this model took about 8 hours of GPU time with 16GB GPU RAM, provided over Kaggle. We have then saved the trained model to file - checkpoint.pth.tar. The saved model is uploaded and used in the current Space.

As part of the app, we provide provide the following features to the user:

• allow users to upload new images
• Provide 3 example images

As part of the output of the app, bounding boxes are shown around the 20 classes, for which the PASCAL VOC dataset is curated. These classes are listed as follows:

"aeroplane",
"bicycle",
"bird",
"boat",
"bottle",
"bus",
"car",
"cat",
"chair",
"cow",
"diningtable",
"dog",
"horse",
"motorbike",
"person",
"pottedplant",
"sheep",
"sofa",
"train",
"tvmonitor"

---