Gureumi commited on
Commit
1bdb27b
·
verified ·
1 Parent(s): 4383bfb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +34 -9
README.md CHANGED
@@ -1,9 +1,11 @@
1
  ---
2
  license: agpl-3.0
3
  metrics:
4
- - precision
5
- - recall
6
- - accuracy
 
 
7
  base_model:
8
  - Ultralytics/YOLO26
9
  pipeline_tag: object-detection
@@ -27,24 +29,47 @@ This model is a fine-tuned YOLO object detection model specifically trained to a
27
 
28
  - **Train Images**: 13,565
29
  - **Validation Images**: 1,507
30
- - **Image Size (`imgsz`)**: 1920 (match for standard VRChat screenshot resolution)
31
  - **Data Source**: Photos captured using the in-game camera over 7 years of VRChat gameplay, along with frame data extracted from 1,707 hours of recorded video footage.
32
 
33
  ### Data Filtering & Preprocessing
34
  - **Included**: Humanoid 3D avatars only.
35
- - **Excluded**: Non-humanoid avatars, Fallback avatars, and Impostor avatars.
36
 
37
  ## Training Configuration
38
  - **Task**: Object Detection
39
- - **Base Framework**: Ultralytics YOLO
40
- - **Input Resolution**: 1920x1920
41
 
42
  ## Usage
43
 
44
  ```python
45
  from ultralytics import YOLO
46
 
47
- model = YOLO("model.pt")
48
  results = model.predict(source="vrchat_screenshot.png", imgsz=1920)
49
  results[0].show()
50
- ```
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: agpl-3.0
3
  metrics:
4
+ - map50
5
+ - map50-95
6
+ - precision
7
+ - recall
8
+ - f1
9
  base_model:
10
  - Ultralytics/YOLO26
11
  pipeline_tag: object-detection
 
29
 
30
  - **Train Images**: 13,565
31
  - **Validation Images**: 1,507
32
+ - **Image Size (`imgsz`)**: 1920 (Source is 1920x1080 ~ 15360x8360)
33
  - **Data Source**: Photos captured using the in-game camera over 7 years of VRChat gameplay, along with frame data extracted from 1,707 hours of recorded video footage.
34
 
35
  ### Data Filtering & Preprocessing
36
  - **Included**: Humanoid 3D avatars only.
37
+ - **Excluded**: Non-humanoid avatars, Fallback avatars, default robot avatar, and Impostor avatars.
38
 
39
  ## Training Configuration
40
  - **Task**: Object Detection
41
+ - **Base Framework**: Ultralytics YOLO26m
42
+ - **Input Resolution**: 1920
43
 
44
  ## Usage
45
 
46
  ```python
47
  from ultralytics import YOLO
48
 
49
+ model = YOLO("yolo26m_vrchat_avatar.pt")
50
  results = model.predict(source="vrchat_screenshot.png", imgsz=1920)
51
  results[0].show()
52
+ ```
53
+
54
+ ## Performance
55
+
56
+ | Metric | Score (%) |
57
+ | ---------- | --------------- |
58
+ | mAP@50 | 96.30 |
59
+ | mAP@50-95 | 81.30 |
60
+ | Precision | 96.50 |
61
+ | Recall | 92.80 |
62
+ | F1 Score | 94.61 |
63
+ | Parameters | 20.35M |
64
+ | FLOPs | 68.1B |
65
+ | Inference | 4.7ms (NVIDIA H100) |
66
+
67
+ ## Per-Class Performance
68
+
69
+ | Class | Images | Instances | Precision (%) | Recall (%) | mAP@50 (%) | mAP@50-95 (%) |
70
+ | ------- | ------ | --------- | ------------- | ---------- | ---------- | ------------- |
71
+ | avatar | 1455 | 2777 | 95.90 | 90.80 | 95.80 | 79.80 |
72
+ | nametag | 121 | 268 | 97.10 | 94.80 | 96.80 | 82.80 |
73
+ | all | 1507 | 3045 | 96.50 | 92.80 | 96.30 | 81.30 |
74
+
75
+ ---