File size: 1,199 Bytes
da6e9f3 063ec0a da6e9f3 | 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 | # VR Mini Project 1 - Best Models Inference Payload
This repository contains the final, best-performing models and the standardized inference script (`predictor.py`) for the Visual Recognition Apparel Detection and Segmentation project.
## Models Included
Based on evaluation metrics, the following architectures were selected:
1. **Classification (Task 3.1):** `cls.pth` — Fine-tuned **ResNet-50**
2. **Detection & Segmentation (Task 3.2):** `seg.pt` — Fine-tuned **YOLOv8-seg**
## Categories Handled
Both models are configured to predict the following 5 canonical categories:
* `short sleeve top`
* `long sleeve top`
* `trousers`
* `shorts`
* `skirt`
## File Structure
```text
.
├── model_files/
│ ├── cls.pth # ResNet-50 weights
│ └── seg.pt # YOLOv8 weights
├── predictor.py # Standardized inference wrapper
├── requirements.txt # Dependencies (torch, torchvision, ultralytics, etc.)
└── validator_local.py# Local validation script for format checking
```
## Team Members
* **Dhruv Ramesh Joshi** – IMT2023032
* **Ankith Kini** – IMT2023075
* **Arnav Oruganty** – IMT2023078
* **Mithilesh Sai Yechuri** – IMT2023507 |