|
|
--- |
|
|
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" |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
--- |
|
|
|
|
|
|
|
|
|
