diffusion-wikiart / mapping.py
kaupane's picture
Update mapping.py
e8445c4 verified
genre_mapping = {
'None': 0,
'abstract': 1,
'advertisement': 2,
'allegorical painting': 3,
'animal painting': 4,
'battle painting': 5,
'bijinga': 6,
'bird-and-flower painting': 7,
'calligraphy': 8,
'capriccio': 9,
'caricature': 10,
'cityscape': 11,
'cloudscape': 12,
'design': 13,
'figurative': 14,
'flower painting': 15,
'genre painting': 16,
'history painting': 17,
'illustration': 18,
'interior': 19,
'landscape': 20,
'literary painting': 21,
'marina': 22,
'miniature': 23,
'mythological painting': 24,
'nude painting (nu)': 25,
'panorama': 26,
'pastorale': 27,
'portrait': 28,
'poster': 29,
'quadratura': 30,
'religious painting': 31,
'self-portrait': 32,
'shan shui': 33,
'sketch and study': 34,
'still life': 35,
'symbolic painting': 36,
'tessellation': 37,
'urushi-e': 38,
'vanitas': 39,
'veduta': 40,
'wildlife painting': 41,
'yakusha-e': 42
}
style_mapping = {
'Abstract Art': 0,
'Abstract Expressionism': 1,
'Academicism': 2,
'Action painting': 3,
'American Realism': 4,
'Analytical Cubism': 5,
'Analytical\xa0Realism': 6,
'Art Brut': 7,
'Art Deco': 8,
'Art Informel': 9,
'Art Nouveau (Modern)': 10,
'Automatic Painting': 11,
'Baroque': 12,
'Biedermeier': 13,
'Byzantine': 14,
'Cartographic Art': 15,
'Classicism': 16,
'Cloisonism': 17,
'Color Field Painting': 18,
'Conceptual Art': 19,
'Concretism': 20,
'Constructivism': 21,
'Contemporary Realism': 22,
'Costumbrismo': 23,
'Cubism': 24,
'Cubo-Expressionism': 25,
'Cubo-Futurism': 26,
'Dada': 27,
'Divisionism': 28,
'Early Renaissance': 29,
'Environmental (Land) Art': 30,
'Existential Art': 31,
'Expressionism': 32,
'Fantastic Realism': 33,
'Fauvism': 34,
'Feminist Art': 35,
'Figurative Expressionism': 36,
'Futurism': 37,
'Gongbi': 38,
'Gothic': 39,
'Hard Edge Painting': 40,
'High Renaissance': 41,
'Hyper-Realism': 42,
'Ilkhanid': 43,
'Impressionism': 44,
'Indian Space painting': 45,
'Ink and wash painting': 46,
'International Gothic': 47,
'Intimism': 48,
'Japonism': 49,
'Joseon Dynasty': 50,
'Kinetic Art': 51,
'Kitsch': 52,
'Lettrism': 53,
'Light and Space': 54,
'Luminism': 55,
'Lyrical Abstraction': 56,
'Magic Realism': 57,
'Mail Art': 58,
'Mannerism (Late Renaissance)': 59,
'Mechanistic Cubism': 60,
'Metaphysical art': 61,
'Minimalism': 62,
'Miserabilism': 63,
'Modernismo': 64,
'Mosan art': 65,
'Muralism': 66,
'Nanga (Bunjinga)': 67,
'Nats-Taliq': 68,
'Native Art': 69,
'Naturalism': 70,
'Naïve Art (Primitivism)': 71,
'Neo-Byzantine': 72,
'Neo-Concretism': 73,
'Neo-Dada': 74,
'Neo-Expressionism': 75,
'Neo-Figurative Art': 76,
'Neo-Rococo': 77,
'Neo-Romanticism': 78,
'Neo-baroque': 79,
'Neoclassicism': 80,
'Neoplasticism': 81,
'New Casualism': 82,
'New European Painting': 83,
'New Realism': 84,
'Nihonga': 85,
'None': 86,
'Northern Renaissance': 87,
'Nouveau Réalisme': 88,
'Op Art': 89,
'Orientalism': 90,
'Orphism': 91,
'Ottoman Period': 92,
'Outsider art': 93,
'Perceptism': 94,
'Photorealism': 95,
'Pointillism': 96,
'Pop Art': 97,
'Post-Impressionism': 98,
'Post-Minimalism': 99,
'Post-Painterly Abstraction': 100,
'Poster Art Realism': 101,
'Precisionism': 102,
'Primitivism': 103,
'Proto Renaissance': 104,
'Purism': 105,
'Rayonism': 106,
'Realism': 107,
'Regionalism': 108,
'Renaissance': 109,
'Rococo': 110,
'Romanesque': 111,
'Romanticism': 112,
'Safavid Period': 113,
'Shin-hanga': 114,
'Social Realism': 115,
'Socialist Realism': 116,
'Spatialism': 117,
'Spectralism': 118,
'Street art': 119,
'Suprematism': 120,
'Surrealism': 121,
'Symbolism': 122,
'Synchromism': 123,
'Synthetic Cubism': 124,
'Synthetism': 125,
'Sōsaku hanga': 126,
'Tachisme': 127,
'Tenebrism': 128,
'Timurid Period': 129,
'Tonalism': 130,
'Transautomatism': 131,
'Tubism': 132,
'Ukiyo-e': 133,
'Verism': 134,
'Yamato-e': 135,
'Zen': 136
}
reverse_genre_mapping = {v: k for k, v in genre_mapping.items()}
reverse_style_mapping = {v: k for k, v in style_mapping.items()}
reduced_genre_mapping = {
'abstract': 1,
'capriccio': 9,
'cityscape': 11,
'cloudscape': 12,
'figurative': 14,
'flower painting': 15,
'genre painting': 16,
'interior': 19,
'landscape': 20,
'marina': 22,
'panorama': 26,
'pastorale': 27,
'portrait': 28,
'quadratura': 30,
'shan shui': 33,
'sketch and study': 34,
'still life': 35,
'symbolic painting': 36,
'tesselation': 37,
'veduta': 40
}
reduced_style_mapping = {
'abstract expressionism': 1,
'art deco': 8,
'art nouveau': 10,
'baroque': 12,
'conceptual art': 19,
'expressionism': 32,
'gongbi': 38,
'gothic': 39,
'impressionism': 44,
'luminism': 55,
'minimalism': 62,
'modernism': 64,
'neoclassicism': 80,
'pop-art': 97,
'post-impressionism': 98,
'renaissance': 109,
'realism': 107,
'rococo': 110,
'romanticism': 112,
'surrealism': 121,
'symbolism': 122
}
reverse_reduced_genre_mapping = {v: k for k, v in reduced_genre_mapping.items()}
reverse_reduced_style_mapping = {v: k for k, v in reduced_style_mapping.items()}