sugarquark/pearl

Classification model

Inference

aimv2.forward(pixel_values, return_dict=True).last_hidden_state
pooled = hidden_states.mean(dim=1)
logits = classifier(pooled)

Categories

[1] Art Nouveau
[2] Baroque
[3] Contemporary Realism
[4] Expressionism
[5] Early Renaissance
[6] High Renaissance
[7] Impressionism
[8] Neoclassicism
[9] Northern Renaissance
[10] Photo
[11] Pointillism
[12] Post-Impressionism
[13] Realism
[14] Renaissance
[15] Rococo
[16] Romanticism
[19] 4o
[20] Cityscape
[21] Figurative
[22] Flower painting
[23] Genre painting
[24] Landscape
[25] Nude painting
[26] Portrait
[27] Religious painting
[28] Self-portrait
[29] Still life
[31] Hat
[34] Top
[37] Dress
[39] Shoes
[46] Bag
[49] Outer