parrvv commited on
Commit
865339f
·
verified ·
1 Parent(s): f280481

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +27 -3
README.md CHANGED
@@ -1,3 +1,27 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - vision
5
+ - image-classification
6
+ - instance-segmentation
7
+ - object-detection
8
+ - pytorch
9
+ - ultralytics
10
+ - deepfashion2
11
+ ---
12
+
13
+ # Clothing Classification & Instance Segmentation
14
+
15
+ This repository contains the best-performing models for VR Mini Project 1.
16
+
17
+ 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`.
18
+
19
+ ## Models Included:
20
+ * **Task 3.1 (Classification):** ResNet50 (`cls.pth`) - Multi-label image classification.
21
+ * **Task 3.2 (Detection & Segmentation):** YOLOv8 (`seg.pt`) - Object detection and instance segmentation.
22
+
23
+ ## Installation
24
+ To run inference, install the required packages:
25
+
26
+ ```bash
27
+ pip install torch torchvision ultralytics Pillow opencv-python-headless huggingface_hub