jdhr/VRMP1_IMT2023032_IMT2023075_IMT2023078_IMT2023507

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

.
├── 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