---
license: mit
tags:
- vision
- image-classification
- instance-segmentation
- object-detection
- pytorch
- ultralytics
- deepfashion2
---

# 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:

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