Added Open Images helper 🧰

Author: manolo-lolo

taming/data/open_images_helper.py

@@ -0,0 +1,379 @@
+open_images_unify_categories_for_coco = {
+    '/m/03bt1vf': '/m/01g317',
+    '/m/04yx4': '/m/01g317',
+    '/m/05r655': '/m/01g317',
+    '/m/01bl7v': '/m/01g317',
+    '/m/0cnyhnx': '/m/01xq0k1',
+    '/m/01226z': '/m/018xm',
+    '/m/05ctyq': '/m/018xm',
+    '/m/058qzx': '/m/04ctx',
+    '/m/06pcq': '/m/0l515',
+    '/m/03m3pdh': '/m/02crq1',
+    '/m/046dlr': '/m/01x3z',
+    '/m/0h8mzrc': '/m/01x3z',
+}
+
+
+top_300_classes_plus_coco_compatibility = [
+    ('Man', 1060962),
+    ('Clothing', 986610),
+    ('Tree', 748162),
+    ('Woman', 611896),
+    ('Person', 610294),
+    ('Human face', 442948),
+    ('Girl', 175399),
+    ('Building', 162147),
+    ('Car', 159135),
+    ('Plant', 155704),
+    ('Human body', 137073),
+    ('Flower', 133128),
+    ('Window', 127485),
+    ('Human arm', 118380),
+    ('House', 114365),
+    ('Wheel', 111684),
+    ('Suit', 99054),
+    ('Human hair', 98089),
+    ('Human head', 92763),
+    ('Chair', 88624),
+    ('Boy', 79849),
+    ('Table', 73699),
+    ('Jeans', 57200),
+    ('Tire', 55725),
+    ('Skyscraper', 53321),
+    ('Food', 52400),
+    ('Footwear', 50335),
+    ('Dress', 50236),
+    ('Human leg', 47124),
+    ('Toy', 46636),
+    ('Tower', 45605),
+    ('Boat', 43486),
+    ('Land vehicle', 40541),
+    ('Bicycle wheel', 34646),
+    ('Palm tree', 33729),
+    ('Fashion accessory', 32914),
+    ('Glasses', 31940),
+    ('Bicycle', 31409),
+    ('Furniture', 30656),
+    ('Sculpture', 29643),
+    ('Bottle', 27558),
+    ('Dog', 26980),
+    ('Snack', 26796),
+    ('Human hand', 26664),
+    ('Bird', 25791),
+    ('Book', 25415),
+    ('Guitar', 24386),
+    ('Jacket', 23998),
+    ('Poster', 22192),
+    ('Dessert', 21284),
+    ('Baked goods', 20657),
+    ('Drink', 19754),
+    ('Flag', 18588),
+    ('Houseplant', 18205),
+    ('Tableware', 17613),
+    ('Airplane', 17218),
+    ('Door', 17195),
+    ('Sports uniform', 17068),
+    ('Shelf', 16865),
+    ('Drum', 16612),
+    ('Vehicle', 16542),
+    ('Microphone', 15269),
+    ('Street light', 14957),
+    ('Cat', 14879),
+    ('Fruit', 13684),
+    ('Fast food', 13536),
+    ('Animal', 12932),
+    ('Vegetable', 12534),
+    ('Train', 12358),
+    ('Horse', 11948),
+    ('Flowerpot', 11728),
+    ('Motorcycle', 11621),
+    ('Fish', 11517),
+    ('Desk', 11405),
+    ('Helmet', 10996),
+    ('Truck', 10915),
+    ('Bus', 10695),
+    ('Hat', 10532),
+    ('Auto part', 10488),
+    ('Musical instrument', 10303),
+    ('Sunglasses', 10207),
+    ('Picture frame', 10096),
+    ('Sports equipment', 10015),
+    ('Shorts', 9999),
+    ('Wine glass', 9632),
+    ('Duck', 9242),
+    ('Wine', 9032),
+    ('Rose', 8781),
+    ('Tie', 8693),
+    ('Butterfly', 8436),
+    ('Beer', 7978),
+    ('Cabinetry', 7956),
+    ('Laptop', 7907),
+    ('Insect', 7497),
+    ('Goggles', 7363),
+    ('Shirt', 7098),
+    ('Dairy Product', 7021),
+    ('Marine invertebrates', 7014),
+    ('Cattle', 7006),
+    ('Trousers', 6903),
+    ('Van', 6843),
+    ('Billboard', 6777),
+    ('Balloon', 6367),
+    ('Human nose', 6103),
+    ('Tent', 6073),
+    ('Camera', 6014),
+    ('Doll', 6002),
+    ('Coat', 5951),
+    ('Mobile phone', 5758),
+    ('Swimwear', 5729),
+    ('Strawberry', 5691),
+    ('Stairs', 5643),
+    ('Goose', 5599),
+    ('Umbrella', 5536),
+    ('Cake', 5508),
+    ('Sun hat', 5475),
+    ('Bench', 5310),
+    ('Bookcase', 5163),
+    ('Bee', 5140),
+    ('Computer monitor', 5078),
+    ('Hiking equipment', 4983),
+    ('Office building', 4981),
+    ('Coffee cup', 4748),
+    ('Curtain', 4685),
+    ('Plate', 4651),
+    ('Box', 4621),
+    ('Tomato', 4595),
+    ('Coffee table', 4529),
+    ('Office supplies', 4473),
+    ('Maple', 4416),
+    ('Muffin', 4365),
+    ('Cocktail', 4234),
+    ('Castle', 4197),
+    ('Couch', 4134),
+    ('Pumpkin', 3983),
+    ('Computer keyboard', 3960),
+    ('Human mouth', 3926),
+    ('Christmas tree', 3893),
+    ('Mushroom', 3883),
+    ('Swimming pool', 3809),
+    ('Pastry', 3799),
+    ('Lavender (Plant)', 3769),
+    ('Football helmet', 3732),
+    ('Bread', 3648),
+    ('Traffic sign', 3628),
+    ('Common sunflower', 3597),
+    ('Television', 3550),
+    ('Bed', 3525),
+    ('Cookie', 3485),
+    ('Fountain', 3484),
+    ('Paddle', 3447),
+    ('Bicycle helmet', 3429),
+    ('Porch', 3420),
+    ('Deer', 3387),
+    ('Fedora', 3339),
+    ('Canoe', 3338),
+    ('Carnivore', 3266),
+    ('Bowl', 3202),
+    ('Human eye', 3166),
+    ('Ball', 3118),
+    ('Pillow', 3077),
+    ('Salad', 3061),
+    ('Beetle', 3060),
+    ('Orange', 3050),
+    ('Drawer', 2958),
+    ('Platter', 2937),
+    ('Elephant', 2921),
+    ('Seafood', 2921),
+    ('Monkey', 2915),
+    ('Countertop', 2879),
+    ('Watercraft', 2831),
+    ('Helicopter', 2805),
+    ('Kitchen appliance', 2797),
+    ('Personal flotation device', 2781),
+    ('Swan', 2739),
+    ('Lamp', 2711),
+    ('Boot', 2695),
+    ('Bronze sculpture', 2693),
+    ('Chicken', 2677),
+    ('Taxi', 2643),
+    ('Juice', 2615),
+    ('Cowboy hat', 2604),
+    ('Apple', 2600),
+    ('Tin can', 2590),
+    ('Necklace', 2564),
+    ('Ice cream', 2560),
+    ('Human beard', 2539),
+    ('Coin', 2536),
+    ('Candle', 2515),
+    ('Cart', 2512),
+    ('High heels', 2441),
+    ('Weapon', 2433),
+    ('Handbag', 2406),
+    ('Penguin', 2396),
+    ('Rifle', 2352),
+    ('Violin', 2336),
+    ('Skull', 2304),
+    ('Lantern', 2285),
+    ('Scarf', 2269),
+    ('Saucer', 2225),
+    ('Sheep', 2215),
+    ('Vase', 2189),
+    ('Lily', 2180),
+    ('Mug', 2154),
+    ('Parrot', 2140),
+    ('Human ear', 2137),
+    ('Sandal', 2115),
+    ('Lizard', 2100),
+    ('Kitchen & dining room table', 2063),
+    ('Spider', 1977),
+    ('Coffee', 1974),
+    ('Goat', 1926),
+    ('Squirrel', 1922),
+    ('Cello', 1913),
+    ('Sushi', 1881),
+    ('Tortoise', 1876),
+    ('Pizza', 1870),
+    ('Studio couch', 1864),
+    ('Barrel', 1862),
+    ('Cosmetics', 1841),
+    ('Moths and butterflies', 1841),
+    ('Convenience store', 1817),
+    ('Watch', 1792),
+    ('Home appliance', 1786),
+    ('Harbor seal', 1780),
+    ('Luggage and bags', 1756),
+    ('Vehicle registration plate', 1754),
+    ('Shrimp', 1751),
+    ('Jellyfish', 1730),
+    ('French fries', 1723),
+    ('Egg (Food)', 1698),
+    ('Football', 1697),
+    ('Musical keyboard', 1683),
+    ('Falcon', 1674),
+    ('Candy', 1660),
+    ('Medical equipment', 1654),
+    ('Eagle', 1651),
+    ('Dinosaur', 1634),
+    ('Surfboard', 1630),
+    ('Tank', 1628),
+    ('Grape', 1624),
+    ('Lion', 1624),
+    ('Owl', 1622),
+    ('Ski', 1613),
+    ('Waste container', 1606),
+    ('Frog', 1591),
+    ('Sparrow', 1585),
+    ('Rabbit', 1581),
+    ('Pen', 1546),
+    ('Sea lion', 1537),
+    ('Spoon', 1521),
+    ('Sink', 1512),
+    ('Teddy bear', 1507),
+    ('Bull', 1495),
+    ('Sofa bed', 1490),
+    ('Dragonfly', 1479),
+    ('Brassiere', 1478),
+    ('Chest of drawers', 1472),
+    ('Aircraft', 1466),
+    ('Human foot', 1463),
+    ('Pig', 1455),
+    ('Fork', 1454),
+    ('Antelope', 1438),
+    ('Tripod', 1427),
+    ('Tool', 1424),
+    ('Cheese', 1422),
+    ('Lemon', 1397),
+    ('Hamburger', 1393),
+    ('Dolphin', 1390),
+    ('Mirror', 1390),
+    ('Marine mammal', 1387),
+    ('Giraffe', 1385),
+    ('Snake', 1368),
+    ('Gondola', 1364),
+    ('Wheelchair', 1360),
+    ('Piano', 1358),
+    ('Cupboard', 1348),
+    ('Banana', 1345),
+    ('Trumpet', 1335),
+    ('Lighthouse', 1333),
+    ('Invertebrate', 1317),
+    ('Carrot', 1268),
+    ('Sock', 1260),
+    ('Tiger', 1241),
+    ('Camel', 1224),
+    ('Parachute', 1224),
+    ('Bathroom accessory', 1223),
+    ('Earrings', 1221),
+    ('Headphones', 1218),
+    ('Skirt', 1198),
+    ('Skateboard', 1190),
+    ('Sandwich', 1148),
+    ('Saxophone', 1141),
+    ('Goldfish', 1136),
+    ('Stool', 1104),
+    ('Traffic light', 1097),
+    ('Shellfish', 1081),
+    ('Backpack', 1079),
+    ('Sea turtle', 1078),
+    ('Cucumber', 1075),
+    ('Tea', 1051),
+    ('Toilet', 1047),
+    ('Roller skates', 1040),
+    ('Mule', 1039),
+    ('Bust', 1031),
+    ('Broccoli', 1030),
+    ('Crab', 1020),
+    ('Oyster', 1019),
+    ('Cannon', 1012),
+    ('Zebra', 1012),
+    ('French horn', 1008),
+    ('Grapefruit', 998),
+    ('Whiteboard', 997),
+    ('Zucchini', 997),
+    ('Crocodile', 992),
+
+    ('Clock', 960),
+    ('Wall clock', 958),
+
+    ('Doughnut', 869),
+    ('Snail', 868),
+
+    ('Baseball glove', 859),
+
+    ('Panda', 830),
+    ('Tennis racket', 830),
+
+    ('Pear', 652),
+
+    ('Bagel', 617),
+    ('Oven', 616),
+    ('Ladybug', 615),
+    ('Shark', 615),
+    ('Polar bear', 614),
+    ('Ostrich', 609),
+
+    ('Hot dog', 473),
+    ('Microwave oven', 467),
+    ('Fire hydrant', 20),
+    ('Stop sign', 20),
+    ('Parking meter', 20),
+    ('Bear', 20),
+    ('Flying disc', 20),
+    ('Snowboard', 20),
+    ('Tennis ball', 20),
+    ('Kite', 20),
+    ('Baseball bat', 20),
+    ('Kitchen knife', 20),
+    ('Knife', 20),
+    ('Submarine sandwich', 20),
+    ('Computer mouse', 20),
+    ('Remote control', 20),
+    ('Toaster', 20),
+    ('Sink', 20),
+    ('Refrigerator', 20),
+    ('Alarm clock', 20),
+    ('Wall clock', 20),
+    ('Scissors', 20),
+    ('Hair dryer', 20),
+    ('Toothbrush', 20),
+    ('Suitcase', 20)
+]