Hugging Face's logo

UmaDiffusion
/
ULTIMA-YOLOv9

Object Detection
YOLO
YOLOv9
Model card Files Community
BootsofLagrangian commited on
Commit
306ac34
•
1 Parent(s): 1bb8c6b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +153 -5
README.md CHANGED
@@ -4,16 +4,26 @@ license_name: umamusume-derivativework-guidelines
4
  license_link: https://umamusume.jp/derivativework_guidelines/
5
  ---
6
 
7
- This is the model repository for ULTIMA-YOLOv9, containing the following checkpoints:
8
- - YOLO9-E
9
-
10
  # About **ULTIMA-YOLO** models
11
 
 
 
 
 
 
 
 
 
 
 
12
  This is a part of [ULTIMA](https://huggingface.co/datasets/UmaDiffusion/ULTIMA) project.
13
 
14
- ULTIMA-YOLOv9 model is a facial detection model for Uma Musumes in illustrations and based on [yolov9-e](https://arxiv.org/abs/2402.13616) and [ULTIMA-YOLO dataset](https://huggingface.co/datasets/UmaDiffusion/ULTIMA-YOLO)
15
 
16
- [ULTIMA Dataset](https://huggingface.co/datasets/UmaDiffusion/ULTIMA) is **U**ma Musume **L**abeled **T**ext-**I**mage **M**ultimodal **A**lignment Dataset.
 
 
 
17
 
18
 
19
  ### How to Use
@@ -67,7 +77,145 @@ def predict(image_path, weights='ultima_yolov9-e.pt', imgsz=640, conf_thres=0.1,
67
  # Apply NMS
68
  pred = non_max_suppression(pred[0][0], conf_thres, iou_thres, classes=None, max_det=1000)
69
  ```
 
70
 
 
 
 
71
 
72
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
73
 
 
4
  license_link: https://umamusume.jp/derivativework_guidelines/
5
  ---
6
 
 
 
 
7
  # About **ULTIMA-YOLO** models
8
 
9
+ <div style="display: flex;">
10
+ <img src="./3rd.jpg" style="width: 1200px; height: auto; margin-right: 10px;">
11
+ </div>
12
+
13
+ <div style="display: flex;">
14
+ <img src="./0th.png" style="width: 400px; height: auto; margin-right: 10px;">
15
+ <img src="./1st.jpg" style="width: 400px; height: auto; margin-right: 10px;">
16
+ <img src="./2nd.png" style="width: 400px; height: auto; margin-right: 10px;">
17
+ </div>
18
+
19
  This is a part of [ULTIMA](https://huggingface.co/datasets/UmaDiffusion/ULTIMA) project.
20
 
21
+ ULTIMA is **U**ma Musume **L**abeled **T**ext-**I**mage **M**ultimodal **A**lignment.
22
 
23
+ ULTIMA-YOLOv9 model is a facial detection model for Uma Musumes in illustrations and based on [yolov9](https://arxiv.org/abs/2402.13616)-e and [ULTIMA-YOLO dataset](https://huggingface.co/datasets/UmaDiffusion/ULTIMA-YOLO)
24
+
25
+ This is the model repository for ULTIMA-YOLOv9, containing the following checkpoints:
26
+ - YOLO9-E
27
 
28
 
29
  ### How to Use
 
77
  # Apply NMS
78
  pred = non_max_suppression(pred[0][0], conf_thres, iou_thres, classes=None, max_det=1000)
79
  ```
80
+ or use `detect_dual.py` in yolov9 repo.
81
 
82
+ ```bash
83
+ python ./detect_dual.py --source [image_path] --device 0 --img 1280 --weights './ultima_yolov9.pt' --name ultima_yolov9_1280_detect
84
+ ```
85
 
86
 
87
+ # Statistics
88
+
89
+ - Train: 3,991 images
90
+ - Val: 399 images
91
+
92
+ | Character Name | # in Train | # in Val | Precision | Recall | mAP50 | mAP50-95 |
93
+ |:------------------:|:---:|:---:|:---:|:---:|:---:|:---:|
94
+ | Agnes Tachyon | 187 | 35 | 0.957 | 0.886 | 0.961 | 0.765 |
95
+ | Air Groove | 87 | 12 | 1 | 0.835 | 0.933 | 0.713 |
96
+ | Air Shakur | 75 | 12 | 0.986 | 1 | 0.995 | 0.909 |
97
+ | Akikawa Yayoi | 25 | 3 | 1 | 0.693 | 0.995 | 0.648 |
98
+ | Admire Vega | 74 | 16 | 1 | 0.754 | 0.894 | 0.707 |
99
+ | Agnes Digital | 50 | 6 | 0.992 | 0.833 | 0.972 | 0.803 |
100
+ | Anshinzawa Sasami | 25 | 1 | 0.956 | 1 | 0.995 | 0.796 |
101
+ | Aston Machan | 55 | 3 | 1 | 0.726 | 0.995 | 0.912 |
102
+ | Bamboo Memory | 41 | 3 | 0.97 | 1 | 0.995 | 0.895 |
103
+ | Biko Pegasus | 34 | 3 | 0.972 | 1 | 0.995 | 0.84 |
104
+ | Byerley Turk | 43 | 2 | 0.951 | 1 | 0.995 | 0.855 |
105
+ | Bitter Glace | 24 | 0 | 0.888 | 0.875 | 0.944 | 0.776 |
106
+ | Biwa Hayahide | 52 | 8 | 0.821 | 1 | 0.995 | 0.846 |
107
+ | Copano Rickey | 51 | 5 | 0.969 | 0.667 | 0.864 | 0.69 |
108
+ | Curren Chan | 54 | 9 | 0.996 | 1 | 0.995 | 0.801 |
109
+ | Cheval Grand | 43 | 13 | 0.998 | 1 | 0.995 | 0.783 |
110
+ | Twin Turbo | 120 | 13 | 0.982 | 1 | 0.995 | 0.842 |
111
+ | Daiichi Ruby | 57 | 5 | 0.963 | 1 | 0.995 | 0.949 |
112
+ | Darley Arabian | 48 | 2 | 1 | 0.837 | 0.995 | 0.819 |
113
+ | Daring Tact | 62 | 5 | 0.997 | 1 | 0.995 | 0.841 |
114
+ | Daitaku Helios | 100 | 11 | 1 | 0.903 | 0.961 | 0.787 |
115
+ | Daiwa Scarlet | 114 | 19 | 0.987 | 1 | 0.995 | 0.707 |
116
+ | El Condor Pasa | 65 | 6 | 0.951 | 1 | 0.995 | 0.808 |
117
+ | Eishin Flash | 39 | 5 | 0.853 | 1 | 0.995 | 0.927 |
118
+ | Fuji Kiseki | 48 | 6 | 1 | 0.875 | 0.995 | 0.88 |
119
+ | Fine Motion | 55 | 7 | 0.989 | 0.875 | 0.906 | 0.71 |
120
+ | Gold City | 49 | 8 | 0.942 | 0.938 | 0.991 | 0.81 |
121
+ | Gold Ship | 146 | 16 | 0.858 | 1 | 0.995 | 0.895 |
122
+ | Godolphin Barb | 44 | 2 | 0.84 | 0.833 | 0.851 | 0.659 |
123
+ | Grass Wonder | 74 | 6 | 1 | 0.797 | 0.995 | 0.792 |
124
+ | Hishi Akebono | 39 | 4 | 0.989 | 1 | 0.995 | 0.766 |
125
+ | Hishi Amazon | 46 | 6 | 0.993 | 1 | 0.995 | 0.835 |
126
+ | Hayakawa Tazuna | 34 | 5 | 1 | 0.659 | 0.922 | 0.638 |
127
+ | Hishi Miracle | 52 | 6 | 0.971 | 0.75 | 0.945 | 0.769 |
128
+ | Happy Meek | 51 | 4 | 1 | 0.787 | 0.938 | 0.808 |
129
+ | Hokko Tarumae | 50 | 9 | 1 | 0.678 | 0.995 | 0.76 |
130
+ | Haru Urara | 69 | 9 | 0.986 | 0.917 | 0.989 | 0.747 |
131
+ | Ikuno Dictus | 96 | 12 | 0.873 | 1 | 0.995 | 0.858 |
132
+ | Ines Fujin | 41 | 7 | 0.947 | 1 | 0.995 | 0.898 |
133
+ | Inari One | 46 | 2 | 0.856 | 1 | 0.995 | 0.656 |
134
+ | Jungle Pocket | 53 | 6 | 1 | 0.85 | 0.995 | 0.747 |
135
+ | King Halo | 77 | 6 | 0.975 | 1 | 0.995 | 0.773 |
136
+ | Kashimoto Riko | 34 | 3 | 1 | 0.778 | 0.995 | 0.823 |
137
+ | Kiryuin Aoi | 44 | 4 | 0.997 | 0.895 | 0.929 | 0.712 |
138
+ | Kitasan Black | 116 | 19 | 0.974 | 1 | 0.995 | 0.909 |
139
+ | K.S.Miracle | 48 | 3 | 0.982 | 1 | 0.995 | 0.852 |
140
+ | Katsuragi Ace | 43 | 4 | 0.989 | 1 | 0.995 | 0.881 |
141
+ | Kawakami Princess | 50 | 7 | 0.975 | 1 | 0.995 | 0.841 |
142
+ | Little Cocon | 51 | 3 | 1 | 0.567 | 0.995 | 0.796 |
143
+ | Light Hello | 25 | 2 | 0.993 | 1 | 0.995 | 0.788 |
144
+ | Mr. C.B. | 91 | 13 | 1 | 0.659 | 0.995 | 0.703 |
145
+ | Meisho Doto | 59 | 7 | 0.988 | 1 | 0.995 | 0.782 |
146
+ | Mihono Bourbon | 84 | 13 | 1 | 0.955 | 0.994 | 0.779 |
147
+ | Manhattan Cafe | 144 | 32 | 0.876 | 0.884 | 0.967 | 0.797 |
148
+ | Mejiro Ardan | 58 | 8 | 0.985 | 0.833 | 0.869 | 0.723 |
149
+ | Mejiro Bright | 55 | 6 | 0.987 | 1 | 0.995 | 0.813 |
150
+ | Mejiro Dober | 56 | 5 | 0.981 | 0.933 | 0.972 | 0.785 |
151
+ | Mejiro McQueen | 272 | 30 | 0.98 | 1 | 0.995 | 0.873 |
152
+ | Mejiro Ryan | 43 | 7 | 0.998 | 1 | 0.995 | 0.849 |
153
+ | Matikanefukukitaru | 52 | 7 | 1 | 0.952 | 0.995 | 0.719 |
154
+ | Matikanetannhauser | 87 | 13 | 0.996 | 1 | 0.995 | 0.81 |
155
+ | Mejiro Palmer | 95 | 11 | 0.893 | 1 | 0.929 | 0.822 |
156
+ | Mejiro Ramonu | 52 | 9 | 0.993 | 1 | 0.995 | 0.748 |
157
+ | Maruzensky | 43 | 7 | 0.984 | 1 | 0.995 | 0.684 |
158
+ | Marvelous Sunday | 40 | 6 | 1 | 0.702 | 0.995 | 0.668 |
159
+ | Nakayama Festa | 47 | 7 | 0.992 | 1 | 0.995 | 0.829 |
160
+ | Nice Nature | 96 | 8 | 0.993 | 1 | 0.995 | 0.723 |
161
+ | Narita Brian | 86 | 13 | 0.827 | 1 | 0.962 | 0.778 |
162
+ | Narita Taishin | 55 | 5 | 0.899 | 0.857 | 0.978 | 0.938 |
163
+ | Nishino Flower | 48 | 7 | 0.97 | 1 | 0.995 | 0.72 |
164
+ | Narita Top Road | 50 | 9 | 0.988 | 1 | 0.995 | 0.834 |
165
+ | Oguri Cap | 94 | 10 | 0.997 | 0.92 | 0.945 | 0.744 |
166
+ | Rice Shower | 165 | 25 | 0.992 | 1 | 0.995 | 0.89 |
167
+ | Sakura Bakushin O | 55 | 7 | 1 | 0.949 | 0.995 | 0.795 |
168
+ | Symboli Rudolf | 157 | 17 | 0.987 | 0.889 | 0.975 | 0.748 |
169
+ | Sakura Chiyono O | 48 | 9 | 0.946 | 0.8 | 0.941 | 0.835 |
170
+ | Seiun Sky | 72 | 10 | 0.98 | 1 | 0.995 | 0.842 |
171
+ | Sakura Laurel | 44 | 6 | 0.944 | 1 | 0.995 | 0.895 |
172
+ | Shinko Windy | 46 | 1 | 0.96 | 1 | 0.995 | 0.949 |
173
+ | Seeking the Pearl | 34 | 2 | 0.985 | 1 | 0.995 | 0.844 |
174
+ | Symboli Kris S | 68 | 6 | 0.87 | 0.958 | 0.943 | 0.728 |
175
+ | Smart Falcon | 53 | 7 | 0.976 | 1 | 0.995 | 0.876 |
176
+ | Super Creek | 48 | 4 | 1 | 0.959 | 0.995 | 0.736 |
177
+ | Special Week | 147 | 14 | 1 | 0.975 | 0.995 | 0.777 |
178
+ | Silence Suzuka | 129 | 18 | 0.993 | 1 | 0.995 | 0.84 |
179
+ | Sirius Symboli | 60 | 9 | 0.962 | 1 | 0.995 | 0.849 |
180
+ | Satono Crown | 47 | 2 | 0.993 | 0.75 | 0.925 | 0.746 |
181
+ | Satono Diamond | 79 | 12 | 0.98 | 0.75 | 0.775 | 0.649 |
182
+ | Sweep Tosho | 42 | 4 | 0.951 | 1 | 0.995 | 0.895 |
183
+ | Tap Dance City | 49 | 4 | 0.995 | 1 | 0.995 | 0.832 |
184
+ | Taiki Shuttle | 50 | 7 | 0.883 | 1 | 0.939 | 0.756 |
185
+ | Tokai Teio | 239 | 23 | 0.994 | 1 | 0.995 | 0.56 |
186
+ | Tamamo Cross | 59 | 6 | 1 | 0.86 | 0.99 | 0.748 |
187
+ | T.M. Opera O | 85 | 13 | 0.986 | 1 | 0.995 | 0.838 |
188
+ | Tanino Gimlet | 52 | 6 | 0.986 | 1 | 0.995 | 0.771 |
189
+ | Mayano Top Gun | 70 | 5 | 1 | 0.824 | 0.995 | 0.787 |
190
+ | Tosen Jordan | 68 | 9 | 0.959 | 1 | 0.995 | 0.801 |
191
+ | Tsurumaru Tsuyoshi | 38 | 2 | 0.984 | 1 | 0.995 | 0.736 |
192
+ | Neo Universe | 47 | 5 | 1 | 0.806 | 0.945 | 0.753 |
193
+ | Vodka | 110 | 15 | 0.954 | 1 | 0.995 | 0.895 |
194
+ | Wonder Acute | 53 | 1 | 0.976 | 0.8 | 0.962 | 0.877 |
195
+ | Winning Ticket | 47 | 5 | 0.997 | 1 | 0.995 | 0.889 |
196
+ | Yukino Bijin | 44 | 7 | 1 | 0.965 | 0.995 | 0.904 |
197
+ | Yaeno Muteki | 39 | 5 | 0.975 | 1 | 0.995 | 0.932 |
198
+ | Yamanin Zephyr | 42 | 3 | 0.976 | 0.714 | 0.96 | 0.747 |
199
+ | Zenno Rob Roy | 51 | 7 | 0.958 | 1 | 0.995 | 0.895 |
200
+ | Furioso | 15 | 0 | 0.938 | 1 | 0.995 | 0.995 |
201
+ | Transcend | 40 | 2 | 0.964 | 1 | 0.995 | 0.796 |
202
+ | Espoir City | 30 | 1 | 0.939 | 1 | 0.995 | 0.895 |
203
+ | North Flight | 40 | 2 | 0.946 | 1 | 0.995 | 0.597 |
204
+ | Dantsu Flame | 30 | 1 | 0.878 | 1 | 0.995 | 0.895 |
205
+ | No Reason | 26 | 0 | 0.961 | 0.667 | 0.699 | 0.53 |
206
+ | Still in Love | 28 | 1 | 0.961 | 1 | 0.995 | 0.895 |
207
+ | Samson Big | 25 | 1 | 0.891 | 1 | 0.995 | 0.697 |
208
+ | Sounds of Earth | 53 | 3 | 0.972 | 1 | 0.995 | 0.857 |
209
+ | Royce and Royce | 30 | 2 | 0.942 | 1 | 0.995 | 0.398 |
210
+ | Duramente | 43 | 1 | 0.939 | 1 | 0.995 | 0.895 |
211
+ | Rhein Kraft | 31 | 3 | 0.975 | 1 | 0.995 | 0.799 |
212
+ | Cesario | 37 | 1 | 0.947 | 1 | 0.995 | 0.796 |
213
+ | Air Messiah | 23 | 1 | 0.964 | 1 | 0.995 | 0.927 |
214
+ | Daring Heart | 28 | 0 | 0.961 | 1 | 0.995 | 0.858 |
215
+ | Orfevre | 25 | 3 | 0.947 | 1 | 0.995 | 0.995 |
216
+ | Gentildonna | 40 | 1 | 0.944 | 1 | 0.995 | 0.597 |
217
+ | Win Variation | 21 | 2 | 0.94 | 1 | 0.995 | 0.895 |
218
+ | Venus Paques | 37 | 2 | 0.935 | 1 | 0.995 | 0.796 |
219
+ | Rigantona | 28 | 1 | 0.995 | 1 | 0.995 | 0.91 |
220
+ | Sonon Elfie | 29 | 1 | 0.994 | 1 | 0.995 | 0.815 |
221