parrvv/Clothing-classification-segmentation

Clothing Classification & Instance Segmentation

This repository contains the best-performing models for VR Mini Project 1.

The models are trained on a subset of the DeepFashion2 dataset to recognize 5 clothing categories: short sleeve top, long sleeve top, shorts, trousers, skirt.

Models Included:

• Task 3.1 (Classification): ResNet50 (cls.pth) - Multi-label image classification.
• Task 3.2 (Detection & Segmentation): YOLOv8 (seg.pt) - Object detection and instance segmentation.

Installation

To run inference, install the required packages:

pip install torch torchvision ultralytics Pillow opencv-python-headless huggingface_hub