common.py

LABELS = [
    'Abstract_Expressionism',
    'Action_painting',
    'Analytical_Cubism',
    'Art_Nouveau_Modern',
    'Baroque',
    'Color_Field_Painting',
    'Contemporary_Realism',
    'Cubism',
    'Early_Renaissance',
    'Expressionism',
    'Fauvism',
    'High_Renaissance',
    'Impressionism',
    'Mannerism_Late_Renaissance',
    'Minimalism',
    'Naive_Art_Primitivism',
    'New_Realism',
    'Northern_Renaissance',
    'Pointillism',
    'Pop_Art',
    'Post_Impressionism',
    'Realism',
    'Rococo',
    'Romanticism',
    'Symbolism',
    'Synthetic_Cubism',
    'Ukiyo_e'
]

new_id_to_old_id = {0: 3,
 1: 4,
 2: 7,
 3: 8,
 4: 9,
 5: 12,
 6: 15,
 7: 17,
 8: 19,
 9: 20,
 10: 21,
 11: 22,
 12: 23,
 13: 24}

MODELS = {
    "CLIP ViT-B/32 (27 классов)": ["openai/clip-vit-base-patch32", "model_clip.pt"],
    #  "CLIP ViT-B/32 (14 классов)": ["openai/clip-vit-base-patch32", "model_clip_14_classes.pt"],
    "SigLIP2 ViT-B/16 (27 классов)": ["google/siglip2-base-patch16-224", "model_siglip.pt"],
    "ResNet50 (27 классов)": ["DEFAULT", "model_resnet.pt"],
}