Update README.md
Browse files
README.md
CHANGED
@@ -3,12 +3,172 @@ license: apache-2.0
|
|
3 |
datasets:
|
4 |
- Artificio/WikiArt
|
5 |
library_name: transformers
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
6 |
---
|
7 |
|
|
|
|
|
8 |
# **WikiArt-Style**
|
9 |
|
10 |
> **WikiArt-Style** is a vision model fine-tuned from **google/siglip2-base-patch16-224** using the **SiglipForImageClassification** architecture. It classifies art images into one of 137 painting style categories.
|
11 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
12 |
```py
|
13 |
from datasets import load_dataset
|
14 |
|
|
|
3 |
datasets:
|
4 |
- Artificio/WikiArt
|
5 |
library_name: transformers
|
6 |
+
language:
|
7 |
+
- en
|
8 |
+
base_model:
|
9 |
+
- google/siglip2-base-patch16-224
|
10 |
+
pipeline_tag: image-classification
|
11 |
+
tags:
|
12 |
+
- WikiArt
|
13 |
+
- Style
|
14 |
+
- art
|
15 |
+
- SigLIP2
|
16 |
---
|
17 |
|
18 |
+

|
19 |
+
|
20 |
# **WikiArt-Style**
|
21 |
|
22 |
> **WikiArt-Style** is a vision model fine-tuned from **google/siglip2-base-patch16-224** using the **SiglipForImageClassification** architecture. It classifies art images into one of 137 painting style categories.
|
23 |
|
24 |
+
```
|
25 |
+
Classification Report:
|
26 |
+
precision recall f1-score support
|
27 |
+
|
28 |
+
Abstract Art 0.2784 0.3228 0.2990 979
|
29 |
+
Abstract Expressionism 0.3615 0.5757 0.4441 2074
|
30 |
+
Academicism 0.4410 0.0730 0.1253 972
|
31 |
+
Action painting 0.0000 0.0000 0.0000 98
|
32 |
+
American Realism 0.9592 0.1697 0.2883 277
|
33 |
+
Analytical Cubism 0.2000 0.0115 0.0217 87
|
34 |
+
Analytical Realism 0.6667 0.0225 0.0435 89
|
35 |
+
Art Brut 0.0000 0.0000 0.0000 198
|
36 |
+
Art Deco 0.6178 0.6801 0.6475 644
|
37 |
+
Art Informel 0.3295 0.2486 0.2834 1267
|
38 |
+
Art Nouveau (Modern) 0.5183 0.6289 0.5682 4899
|
39 |
+
Automatic Painting 0.0000 0.0000 0.0000 37
|
40 |
+
Baroque 0.5312 0.6495 0.5845 4400
|
41 |
+
Biedermeier 0.0000 0.0000 0.0000 132
|
42 |
+
Byzantine 0.0000 0.0000 0.0000 77
|
43 |
+
Cartographic Art 0.0000 0.0000 0.0000 10
|
44 |
+
Classicism 0.0000 0.0000 0.0000 223
|
45 |
+
Cloisonnism 0.0000 0.0000 0.0000 172
|
46 |
+
Color Field Painting 0.6947 0.5352 0.6046 910
|
47 |
+
Conceptual Art 0.6667 0.0161 0.0315 124
|
48 |
+
Concretism 0.3973 0.4596 0.4262 581
|
49 |
+
Constructivism 0.3462 0.3022 0.3227 268
|
50 |
+
Contemporary Realism 1.0000 0.0518 0.0985 309
|
51 |
+
Costumbrismo 0.0000 0.0000 0.0000 19
|
52 |
+
Cubism 0.3426 0.6926 0.4584 1747
|
53 |
+
Cubo-Expressionism 0.0000 0.0000 0.0000 22
|
54 |
+
Cubo-Futurism 0.0000 0.0000 0.0000 137
|
55 |
+
Dada 0.5000 0.0098 0.0191 205
|
56 |
+
Divisionism 0.6810 0.2337 0.3480 338
|
57 |
+
Early Renaissance 0.5411 0.6329 0.5834 1351
|
58 |
+
Environmental (Land) Art 0.0000 0.0000 0.0000 1
|
59 |
+
Existential Art 0.0000 0.0000 0.0000 17
|
60 |
+
Expressionism 0.4540 0.5701 0.5055 7013
|
61 |
+
Fantastic Realism 0.0000 0.0000 0.0000 31
|
62 |
+
Fauvism 0.4480 0.1710 0.2475 731
|
63 |
+
Feminist Art 0.0000 0.0000 0.0000 18
|
64 |
+
Figurative Expressionism 0.0000 0.0000 0.0000 19
|
65 |
+
Futurism 0.4717 0.0859 0.1453 291
|
66 |
+
Gongbi 0.0000 0.0000 0.0000 38
|
67 |
+
Gothic 0.0000 0.0000 0.0000 12
|
68 |
+
Hard Edge Painting 0.4458 0.3978 0.4205 372
|
69 |
+
High Renaissance 0.5016 0.3653 0.4227 1314
|
70 |
+
Hyper-Realism 0.0000 0.0000 0.0000 49
|
71 |
+
Ilkhanid 0.0000 0.0000 0.0000 2
|
72 |
+
Impressionism 0.6057 0.7266 0.6607 10643
|
73 |
+
Indian Space painting 0.0000 0.0000 0.0000 25
|
74 |
+
Ink and wash painting 0.7739 0.4459 0.5658 545
|
75 |
+
International Gothic 0.8099 0.5349 0.6443 215
|
76 |
+
Intimism 0.0000 0.0000 0.0000 109
|
77 |
+
Japonism 0.0000 0.0000 0.0000 118
|
78 |
+
Joseon Dynasty 0.0000 0.0000 0.0000 10
|
79 |
+
Kinetic Art 0.0000 0.0000 0.0000 3
|
80 |
+
Kitsch 0.0000 0.0000 0.0000 47
|
81 |
+
Lettrism 0.0000 0.0000 0.0000 33
|
82 |
+
Light and Space 0.0000 0.0000 0.0000 11
|
83 |
+
Luminism 0.8333 0.0519 0.0978 385
|
84 |
+
Lyrical Abstraction 0.4631 0.2060 0.2851 670
|
85 |
+
Magic Realism 0.5883 0.5020 0.5417 1002
|
86 |
+
Mail Art 0.0000 0.0000 0.0000 10
|
87 |
+
Mannerism (Late Renaissance) 0.4875 0.3934 0.4355 1342
|
88 |
+
Mechanistic Cubism 0.0000 0.0000 0.0000 69
|
89 |
+
Metaphysical art 0.5385 0.0729 0.1284 192
|
90 |
+
Minimalism 0.5352 0.5957 0.5638 460
|
91 |
+
Miserablism 0.0000 0.0000 0.0000 8
|
92 |
+
Modernismo 0.0000 0.0000 0.0000 43
|
93 |
+
Mosan art 0.0000 0.0000 0.0000 39
|
94 |
+
Muralism 0.3636 0.0268 0.0500 149
|
95 |
+
Nanga (Bunjinga) 0.0000 0.0000 0.0000 58
|
96 |
+
Nas-Taliq 0.0000 0.0000 0.0000 9
|
97 |
+
Native Art 0.0000 0.0000 0.0000 21
|
98 |
+
Naturalism 0.8459 0.6437 0.7311 435
|
99 |
+
Naïve Art (Primitivism) 0.4897 0.5412 0.5142 2295
|
100 |
+
Neo-Byzantine 0.0000 0.0000 0.0000 18
|
101 |
+
Neo-Concretism 0.0000 0.0000 0.0000 44
|
102 |
+
Neo-Dada 0.0000 0.0000 0.0000 131
|
103 |
+
Neo-Expressionism 0.6763 0.2238 0.3363 420
|
104 |
+
Neo-Figurative Art 0.0000 0.0000 0.0000 27
|
105 |
+
Neo-Rococo 0.0000 0.0000 0.0000 97
|
106 |
+
Neo-Romanticism 0.6762 0.1254 0.2116 566
|
107 |
+
Neo-baroque 0.0000 0.0000 0.0000 105
|
108 |
+
Neoclassicism 0.5868 0.3317 0.4238 2038
|
109 |
+
Neoplasticism 0.8889 0.2775 0.4229 173
|
110 |
+
New Casualism 0.0000 0.0000 0.0000 22
|
111 |
+
New European Painting 0.0000 0.0000 0.0000 25
|
112 |
+
New Realism 0.0000 0.0000 0.0000 329
|
113 |
+
Nihonga 0.0000 0.0000 0.0000 29
|
114 |
+
None 0.3056 0.0112 0.0215 986
|
115 |
+
Northern Renaissance 0.6198 0.6448 0.6321 2379
|
116 |
+
Nouveau Réalisme 0.0000 0.0000 0.0000 142
|
117 |
+
Op Art 0.5120 0.7292 0.6016 528
|
118 |
+
Orientalism 0.0000 0.0000 0.0000 392
|
119 |
+
Orphism 0.6471 0.0444 0.0830 248
|
120 |
+
Ottoman Period 0.8571 0.0833 0.1519 72
|
121 |
+
Outsider art 0.0000 0.0000 0.0000 68
|
122 |
+
Perceptism 0.0000 0.0000 0.0000 6
|
123 |
+
Photorealism 0.0000 0.0000 0.0000 61
|
124 |
+
Pointillism 0.6450 0.6347 0.6398 501
|
125 |
+
Pop Art 0.5580 0.4501 0.4983 791
|
126 |
+
Post-Impressionism 0.4611 0.4200 0.4396 5778
|
127 |
+
Post-Minimalism 0.0000 0.0000 0.0000 31
|
128 |
+
Post-Painterly Abstraction 0.6250 0.0296 0.0565 169
|
129 |
+
Poster Art Realism 0.0000 0.0000 0.0000 43
|
130 |
+
Precisionism 0.7258 0.3169 0.4412 284
|
131 |
+
Primitivism 0.0000 0.0000 0.0000 36
|
132 |
+
Proto Renaissance 0.4342 0.3626 0.3952 273
|
133 |
+
Purism 0.5000 0.1635 0.2464 159
|
134 |
+
Rayonism 0.0000 0.0000 0.0000 6
|
135 |
+
Realism 0.4909 0.6939 0.5750 10523
|
136 |
+
Regionalism 0.7500 0.1776 0.2872 321
|
137 |
+
Renaissance 0.0000 0.0000 0.0000 1
|
138 |
+
Rococo 0.4975 0.5913 0.5404 2733
|
139 |
+
Romanesque 0.0000 0.0000 0.0000 55
|
140 |
+
Romanticism 0.5182 0.5380 0.5279 9285
|
141 |
+
Safavid Period 0.0000 0.0000 0.0000 39
|
142 |
+
Shin-hanga 0.6894 0.7184 0.7036 380
|
143 |
+
Social Realism 0.0000 0.0000 0.0000 305
|
144 |
+
Socialist Realism 0.0000 0.0000 0.0000 95
|
145 |
+
Spatialism 0.0000 0.0000 0.0000 83
|
146 |
+
Spectralism 0.0000 0.0000 0.0000 5
|
147 |
+
Street art 0.0000 0.0000 0.0000 23
|
148 |
+
Suprematism 0.0000 0.0000 0.0000 80
|
149 |
+
Surrealism 0.5011 0.6151 0.5523 4167
|
150 |
+
Symbolism 0.4398 0.3087 0.3627 3476
|
151 |
+
Synchromism 0.0000 0.0000 0.0000 10
|
152 |
+
Synthetic Cubism 0.6914 0.2995 0.4179 187
|
153 |
+
Synthetism 0.0000 0.0000 0.0000 49
|
154 |
+
Sōsaku hanga 0.5753 0.2900 0.3856 369
|
155 |
+
Tachisme 0.4366 0.1422 0.2145 436
|
156 |
+
Tenebrism 0.0000 0.0000 0.0000 221
|
157 |
+
Timurid Period 0.0000 0.0000 0.0000 17
|
158 |
+
Tonalism 0.0000 0.0000 0.0000 202
|
159 |
+
Transautomatism 0.0000 0.0000 0.0000 74
|
160 |
+
Tubism 0.0000 0.0000 0.0000 21
|
161 |
+
Ukiyo-e 0.7822 0.8612 0.8198 1426
|
162 |
+
Verism 0.0000 0.0000 0.0000 84
|
163 |
+
Yamato-e 0.0000 0.0000 0.0000 12
|
164 |
+
Zen 0.6702 0.6702 0.6702 94
|
165 |
+
|
166 |
+
accuracy 0.5108 103250
|
167 |
+
macro avg 0.2664 0.1708 0.1811 103250
|
168 |
+
weighted avg 0.4956 0.5108 0.4822 103250
|
169 |
+
```
|
170 |
+
|
171 |
+
|
172 |
```py
|
173 |
from datasets import load_dataset
|
174 |
|