You can deploy this model and start earning money today!

parrvv/
Clothing-classification-segmentation

Name: Clothing-classification-segmentation
Availability: InStock

ultralytics

vision

image-classification

instance-segmentation

Views

No views yet

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

parrvv/Clothing-classification-segmentation

Clothing Classification & Instance Segmentation

Models Included:

Installation

parrvv/
Clothing-classification-segmentation