You can deploy this model and start earning money today!

jdhr/
VRMP1_IMT2023032_IMT2023075_IMT2023078_IMT2023507

Name: VRMP1_IMT2023032_IMT2023075_IMT2023078_IMT2023507
Availability: InStock

Views

No views yet

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:

Classification (Task 3.1): cls.pth — Fine-tuned ResNet-50
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

jdhr/VRMP1_IMT2023032_IMT2023075_IMT2023078_IMT2023507