| | |
| | import copy |
| | import warnings |
| | from typing import List |
| |
|
| | from mmengine.fileio import get_local_path |
| |
|
| | from mmdet.registry import DATASETS |
| | from .coco import CocoDataset |
| |
|
| |
|
| | @DATASETS.register_module() |
| | class LVISV05Dataset(CocoDataset): |
| | """LVIS v0.5 dataset for detection.""" |
| |
|
| | METAINFO = { |
| | 'classes': |
| | ('acorn', 'aerosol_can', 'air_conditioner', 'airplane', 'alarm_clock', |
| | 'alcohol', 'alligator', 'almond', 'ambulance', 'amplifier', 'anklet', |
| | 'antenna', 'apple', 'apple_juice', 'applesauce', 'apricot', 'apron', |
| | 'aquarium', 'armband', 'armchair', 'armoire', 'armor', 'artichoke', |
| | 'trash_can', 'ashtray', 'asparagus', 'atomizer', 'avocado', 'award', |
| | 'awning', 'ax', 'baby_buggy', 'basketball_backboard', 'backpack', |
| | 'handbag', 'suitcase', 'bagel', 'bagpipe', 'baguet', 'bait', 'ball', |
| | 'ballet_skirt', 'balloon', 'bamboo', 'banana', 'Band_Aid', 'bandage', |
| | 'bandanna', 'banjo', 'banner', 'barbell', 'barge', 'barrel', |
| | 'barrette', 'barrow', 'baseball_base', 'baseball', 'baseball_bat', |
| | 'baseball_cap', 'baseball_glove', 'basket', 'basketball_hoop', |
| | 'basketball', 'bass_horn', 'bat_(animal)', 'bath_mat', 'bath_towel', |
| | 'bathrobe', 'bathtub', 'batter_(food)', 'battery', 'beachball', |
| | 'bead', 'beaker', 'bean_curd', 'beanbag', 'beanie', 'bear', 'bed', |
| | 'bedspread', 'cow', 'beef_(food)', 'beeper', 'beer_bottle', |
| | 'beer_can', 'beetle', 'bell', 'bell_pepper', 'belt', 'belt_buckle', |
| | 'bench', 'beret', 'bib', 'Bible', 'bicycle', 'visor', 'binder', |
| | 'binoculars', 'bird', 'birdfeeder', 'birdbath', 'birdcage', |
| | 'birdhouse', 'birthday_cake', 'birthday_card', 'biscuit_(bread)', |
| | 'pirate_flag', 'black_sheep', 'blackboard', 'blanket', 'blazer', |
| | 'blender', 'blimp', 'blinker', 'blueberry', 'boar', 'gameboard', |
| | 'boat', 'bobbin', 'bobby_pin', 'boiled_egg', 'bolo_tie', 'deadbolt', |
| | 'bolt', 'bonnet', 'book', 'book_bag', 'bookcase', 'booklet', |
| | 'bookmark', 'boom_microphone', 'boot', 'bottle', 'bottle_opener', |
| | 'bouquet', 'bow_(weapon)', 'bow_(decorative_ribbons)', 'bow-tie', |
| | 'bowl', 'pipe_bowl', 'bowler_hat', 'bowling_ball', 'bowling_pin', |
| | 'boxing_glove', 'suspenders', 'bracelet', 'brass_plaque', 'brassiere', |
| | 'bread-bin', 'breechcloth', 'bridal_gown', 'briefcase', |
| | 'bristle_brush', 'broccoli', 'broach', 'broom', 'brownie', |
| | 'brussels_sprouts', 'bubble_gum', 'bucket', 'horse_buggy', 'bull', |
| | 'bulldog', 'bulldozer', 'bullet_train', 'bulletin_board', |
| | 'bulletproof_vest', 'bullhorn', 'corned_beef', 'bun', 'bunk_bed', |
| | 'buoy', 'burrito', 'bus_(vehicle)', 'business_card', 'butcher_knife', |
| | 'butter', 'butterfly', 'button', 'cab_(taxi)', 'cabana', 'cabin_car', |
| | 'cabinet', 'locker', 'cake', 'calculator', 'calendar', 'calf', |
| | 'camcorder', 'camel', 'camera', 'camera_lens', 'camper_(vehicle)', |
| | 'can', 'can_opener', 'candelabrum', 'candle', 'candle_holder', |
| | 'candy_bar', 'candy_cane', 'walking_cane', 'canister', 'cannon', |
| | 'canoe', 'cantaloup', 'canteen', 'cap_(headwear)', 'bottle_cap', |
| | 'cape', 'cappuccino', 'car_(automobile)', 'railcar_(part_of_a_train)', |
| | 'elevator_car', 'car_battery', 'identity_card', 'card', 'cardigan', |
| | 'cargo_ship', 'carnation', 'horse_carriage', 'carrot', 'tote_bag', |
| | 'cart', 'carton', 'cash_register', 'casserole', 'cassette', 'cast', |
| | 'cat', 'cauliflower', 'caviar', 'cayenne_(spice)', 'CD_player', |
| | 'celery', 'cellular_telephone', 'chain_mail', 'chair', |
| | 'chaise_longue', 'champagne', 'chandelier', 'chap', 'checkbook', |
| | 'checkerboard', 'cherry', 'chessboard', |
| | 'chest_of_drawers_(furniture)', 'chicken_(animal)', 'chicken_wire', |
| | 'chickpea', 'Chihuahua', 'chili_(vegetable)', 'chime', 'chinaware', |
| | 'crisp_(potato_chip)', 'poker_chip', 'chocolate_bar', |
| | 'chocolate_cake', 'chocolate_milk', 'chocolate_mousse', 'choker', |
| | 'chopping_board', 'chopstick', 'Christmas_tree', 'slide', 'cider', |
| | 'cigar_box', 'cigarette', 'cigarette_case', 'cistern', 'clarinet', |
| | 'clasp', 'cleansing_agent', 'clementine', 'clip', 'clipboard', |
| | 'clock', 'clock_tower', 'clothes_hamper', 'clothespin', 'clutch_bag', |
| | 'coaster', 'coat', 'coat_hanger', 'coatrack', 'cock', 'coconut', |
| | 'coffee_filter', 'coffee_maker', 'coffee_table', 'coffeepot', 'coil', |
| | 'coin', 'colander', 'coleslaw', 'coloring_material', |
| | 'combination_lock', 'pacifier', 'comic_book', 'computer_keyboard', |
| | 'concrete_mixer', 'cone', 'control', 'convertible_(automobile)', |
| | 'sofa_bed', 'cookie', 'cookie_jar', 'cooking_utensil', |
| | 'cooler_(for_food)', 'cork_(bottle_plug)', 'corkboard', 'corkscrew', |
| | 'edible_corn', 'cornbread', 'cornet', 'cornice', 'cornmeal', 'corset', |
| | 'romaine_lettuce', 'costume', 'cougar', 'coverall', 'cowbell', |
| | 'cowboy_hat', 'crab_(animal)', 'cracker', 'crape', 'crate', 'crayon', |
| | 'cream_pitcher', 'credit_card', 'crescent_roll', 'crib', 'crock_pot', |
| | 'crossbar', 'crouton', 'crow', 'crown', 'crucifix', 'cruise_ship', |
| | 'police_cruiser', 'crumb', 'crutch', 'cub_(animal)', 'cube', |
| | 'cucumber', 'cufflink', 'cup', 'trophy_cup', 'cupcake', 'hair_curler', |
| | 'curling_iron', 'curtain', 'cushion', 'custard', 'cutting_tool', |
| | 'cylinder', 'cymbal', 'dachshund', 'dagger', 'dartboard', |
| | 'date_(fruit)', 'deck_chair', 'deer', 'dental_floss', 'desk', |
| | 'detergent', 'diaper', 'diary', 'die', 'dinghy', 'dining_table', |
| | 'tux', 'dish', 'dish_antenna', 'dishrag', 'dishtowel', 'dishwasher', |
| | 'dishwasher_detergent', 'diskette', 'dispenser', 'Dixie_cup', 'dog', |
| | 'dog_collar', 'doll', 'dollar', 'dolphin', 'domestic_ass', 'eye_mask', |
| | 'doorbell', 'doorknob', 'doormat', 'doughnut', 'dove', 'dragonfly', |
| | 'drawer', 'underdrawers', 'dress', 'dress_hat', 'dress_suit', |
| | 'dresser', 'drill', 'drinking_fountain', 'drone', 'dropper', |
| | 'drum_(musical_instrument)', 'drumstick', 'duck', 'duckling', |
| | 'duct_tape', 'duffel_bag', 'dumbbell', 'dumpster', 'dustpan', |
| | 'Dutch_oven', 'eagle', 'earphone', 'earplug', 'earring', 'easel', |
| | 'eclair', 'eel', 'egg', 'egg_roll', 'egg_yolk', 'eggbeater', |
| | 'eggplant', 'electric_chair', 'refrigerator', 'elephant', 'elk', |
| | 'envelope', 'eraser', 'escargot', 'eyepatch', 'falcon', 'fan', |
| | 'faucet', 'fedora', 'ferret', 'Ferris_wheel', 'ferry', 'fig_(fruit)', |
| | 'fighter_jet', 'figurine', 'file_cabinet', 'file_(tool)', |
| | 'fire_alarm', 'fire_engine', 'fire_extinguisher', 'fire_hose', |
| | 'fireplace', 'fireplug', 'fish', 'fish_(food)', 'fishbowl', |
| | 'fishing_boat', 'fishing_rod', 'flag', 'flagpole', 'flamingo', |
| | 'flannel', 'flash', 'flashlight', 'fleece', 'flip-flop_(sandal)', |
| | 'flipper_(footwear)', 'flower_arrangement', 'flute_glass', 'foal', |
| | 'folding_chair', 'food_processor', 'football_(American)', |
| | 'football_helmet', 'footstool', 'fork', 'forklift', 'freight_car', |
| | 'French_toast', 'freshener', 'frisbee', 'frog', 'fruit_juice', |
| | 'fruit_salad', 'frying_pan', 'fudge', 'funnel', 'futon', 'gag', |
| | 'garbage', 'garbage_truck', 'garden_hose', 'gargle', 'gargoyle', |
| | 'garlic', 'gasmask', 'gazelle', 'gelatin', 'gemstone', 'giant_panda', |
| | 'gift_wrap', 'ginger', 'giraffe', 'cincture', |
| | 'glass_(drink_container)', 'globe', 'glove', 'goat', 'goggles', |
| | 'goldfish', 'golf_club', 'golfcart', 'gondola_(boat)', 'goose', |
| | 'gorilla', 'gourd', 'surgical_gown', 'grape', 'grasshopper', 'grater', |
| | 'gravestone', 'gravy_boat', 'green_bean', 'green_onion', 'griddle', |
| | 'grillroom', 'grinder_(tool)', 'grits', 'grizzly', 'grocery_bag', |
| | 'guacamole', 'guitar', 'gull', 'gun', 'hair_spray', 'hairbrush', |
| | 'hairnet', 'hairpin', 'ham', 'hamburger', 'hammer', 'hammock', |
| | 'hamper', 'hamster', 'hair_dryer', 'hand_glass', 'hand_towel', |
| | 'handcart', 'handcuff', 'handkerchief', 'handle', 'handsaw', |
| | 'hardback_book', 'harmonium', 'hat', 'hatbox', 'hatch', 'veil', |
| | 'headband', 'headboard', 'headlight', 'headscarf', 'headset', |
| | 'headstall_(for_horses)', 'hearing_aid', 'heart', 'heater', |
| | 'helicopter', 'helmet', 'heron', 'highchair', 'hinge', 'hippopotamus', |
| | 'hockey_stick', 'hog', 'home_plate_(baseball)', 'honey', 'fume_hood', |
| | 'hook', 'horse', 'hose', 'hot-air_balloon', 'hotplate', 'hot_sauce', |
| | 'hourglass', 'houseboat', 'hummingbird', 'hummus', 'polar_bear', |
| | 'icecream', 'popsicle', 'ice_maker', 'ice_pack', 'ice_skate', |
| | 'ice_tea', 'igniter', 'incense', 'inhaler', 'iPod', |
| | 'iron_(for_clothing)', 'ironing_board', 'jacket', 'jam', 'jean', |
| | 'jeep', 'jelly_bean', 'jersey', 'jet_plane', 'jewelry', 'joystick', |
| | 'jumpsuit', 'kayak', 'keg', 'kennel', 'kettle', 'key', 'keycard', |
| | 'kilt', 'kimono', 'kitchen_sink', 'kitchen_table', 'kite', 'kitten', |
| | 'kiwi_fruit', 'knee_pad', 'knife', 'knight_(chess_piece)', |
| | 'knitting_needle', 'knob', 'knocker_(on_a_door)', 'koala', 'lab_coat', |
| | 'ladder', 'ladle', 'ladybug', 'lamb_(animal)', 'lamb-chop', 'lamp', |
| | 'lamppost', 'lampshade', 'lantern', 'lanyard', 'laptop_computer', |
| | 'lasagna', 'latch', 'lawn_mower', 'leather', 'legging_(clothing)', |
| | 'Lego', 'lemon', 'lemonade', 'lettuce', 'license_plate', 'life_buoy', |
| | 'life_jacket', 'lightbulb', 'lightning_rod', 'lime', 'limousine', |
| | 'linen_paper', 'lion', 'lip_balm', 'lipstick', 'liquor', 'lizard', |
| | 'Loafer_(type_of_shoe)', 'log', 'lollipop', 'lotion', |
| | 'speaker_(stereo_equipment)', 'loveseat', 'machine_gun', 'magazine', |
| | 'magnet', 'mail_slot', 'mailbox_(at_home)', 'mallet', 'mammoth', |
| | 'mandarin_orange', 'manger', 'manhole', 'map', 'marker', 'martini', |
| | 'mascot', 'mashed_potato', 'masher', 'mask', 'mast', |
| | 'mat_(gym_equipment)', 'matchbox', 'mattress', 'measuring_cup', |
| | 'measuring_stick', 'meatball', 'medicine', 'melon', 'microphone', |
| | 'microscope', 'microwave_oven', 'milestone', 'milk', 'minivan', |
| | 'mint_candy', 'mirror', 'mitten', 'mixer_(kitchen_tool)', 'money', |
| | 'monitor_(computer_equipment) computer_monitor', 'monkey', 'motor', |
| | 'motor_scooter', 'motor_vehicle', 'motorboat', 'motorcycle', |
| | 'mound_(baseball)', 'mouse_(animal_rodent)', |
| | 'mouse_(computer_equipment)', 'mousepad', 'muffin', 'mug', 'mushroom', |
| | 'music_stool', 'musical_instrument', 'nailfile', 'nameplate', |
| | 'napkin', 'neckerchief', 'necklace', 'necktie', 'needle', 'nest', |
| | 'newsstand', 'nightshirt', 'nosebag_(for_animals)', |
| | 'noseband_(for_animals)', 'notebook', 'notepad', 'nut', 'nutcracker', |
| | 'oar', 'octopus_(food)', 'octopus_(animal)', 'oil_lamp', 'olive_oil', |
| | 'omelet', 'onion', 'orange_(fruit)', 'orange_juice', 'oregano', |
| | 'ostrich', 'ottoman', 'overalls_(clothing)', 'owl', 'packet', |
| | 'inkpad', 'pad', 'paddle', 'padlock', 'paintbox', 'paintbrush', |
| | 'painting', 'pajamas', 'palette', 'pan_(for_cooking)', |
| | 'pan_(metal_container)', 'pancake', 'pantyhose', 'papaya', |
| | 'paperclip', 'paper_plate', 'paper_towel', 'paperback_book', |
| | 'paperweight', 'parachute', 'parakeet', 'parasail_(sports)', |
| | 'parchment', 'parka', 'parking_meter', 'parrot', |
| | 'passenger_car_(part_of_a_train)', 'passenger_ship', 'passport', |
| | 'pastry', 'patty_(food)', 'pea_(food)', 'peach', 'peanut_butter', |
| | 'pear', 'peeler_(tool_for_fruit_and_vegetables)', 'pegboard', |
| | 'pelican', 'pen', 'pencil', 'pencil_box', 'pencil_sharpener', |
| | 'pendulum', 'penguin', 'pennant', 'penny_(coin)', 'pepper', |
| | 'pepper_mill', 'perfume', 'persimmon', 'baby', 'pet', 'petfood', |
| | 'pew_(church_bench)', 'phonebook', 'phonograph_record', 'piano', |
| | 'pickle', 'pickup_truck', 'pie', 'pigeon', 'piggy_bank', 'pillow', |
| | 'pin_(non_jewelry)', 'pineapple', 'pinecone', 'ping-pong_ball', |
| | 'pinwheel', 'tobacco_pipe', 'pipe', 'pistol', 'pita_(bread)', |
| | 'pitcher_(vessel_for_liquid)', 'pitchfork', 'pizza', 'place_mat', |
| | 'plate', 'platter', 'playing_card', 'playpen', 'pliers', |
| | 'plow_(farm_equipment)', 'pocket_watch', 'pocketknife', |
| | 'poker_(fire_stirring_tool)', 'pole', 'police_van', 'polo_shirt', |
| | 'poncho', 'pony', 'pool_table', 'pop_(soda)', 'portrait', |
| | 'postbox_(public)', 'postcard', 'poster', 'pot', 'flowerpot', |
| | 'potato', 'potholder', 'pottery', 'pouch', 'power_shovel', 'prawn', |
| | 'printer', 'projectile_(weapon)', 'projector', 'propeller', 'prune', |
| | 'pudding', 'puffer_(fish)', 'puffin', 'pug-dog', 'pumpkin', 'puncher', |
| | 'puppet', 'puppy', 'quesadilla', 'quiche', 'quilt', 'rabbit', |
| | 'race_car', 'racket', 'radar', 'radiator', 'radio_receiver', 'radish', |
| | 'raft', 'rag_doll', 'raincoat', 'ram_(animal)', 'raspberry', 'rat', |
| | 'razorblade', 'reamer_(juicer)', 'rearview_mirror', 'receipt', |
| | 'recliner', 'record_player', 'red_cabbage', 'reflector', |
| | 'remote_control', 'rhinoceros', 'rib_(food)', 'rifle', 'ring', |
| | 'river_boat', 'road_map', 'robe', 'rocking_chair', 'roller_skate', |
| | 'Rollerblade', 'rolling_pin', 'root_beer', |
| | 'router_(computer_equipment)', 'rubber_band', 'runner_(carpet)', |
| | 'plastic_bag', 'saddle_(on_an_animal)', 'saddle_blanket', 'saddlebag', |
| | 'safety_pin', 'sail', 'salad', 'salad_plate', 'salami', |
| | 'salmon_(fish)', 'salmon_(food)', 'salsa', 'saltshaker', |
| | 'sandal_(type_of_shoe)', 'sandwich', 'satchel', 'saucepan', 'saucer', |
| | 'sausage', 'sawhorse', 'saxophone', 'scale_(measuring_instrument)', |
| | 'scarecrow', 'scarf', 'school_bus', 'scissors', 'scoreboard', |
| | 'scrambled_eggs', 'scraper', 'scratcher', 'screwdriver', |
| | 'scrubbing_brush', 'sculpture', 'seabird', 'seahorse', 'seaplane', |
| | 'seashell', 'seedling', 'serving_dish', 'sewing_machine', 'shaker', |
| | 'shampoo', 'shark', 'sharpener', 'Sharpie', 'shaver_(electric)', |
| | 'shaving_cream', 'shawl', 'shears', 'sheep', 'shepherd_dog', |
| | 'sherbert', 'shield', 'shirt', 'shoe', 'shopping_bag', |
| | 'shopping_cart', 'short_pants', 'shot_glass', 'shoulder_bag', |
| | 'shovel', 'shower_head', 'shower_curtain', 'shredder_(for_paper)', |
| | 'sieve', 'signboard', 'silo', 'sink', 'skateboard', 'skewer', 'ski', |
| | 'ski_boot', 'ski_parka', 'ski_pole', 'skirt', 'sled', 'sleeping_bag', |
| | 'sling_(bandage)', 'slipper_(footwear)', 'smoothie', 'snake', |
| | 'snowboard', 'snowman', 'snowmobile', 'soap', 'soccer_ball', 'sock', |
| | 'soda_fountain', 'carbonated_water', 'sofa', 'softball', |
| | 'solar_array', 'sombrero', 'soup', 'soup_bowl', 'soupspoon', |
| | 'sour_cream', 'soya_milk', 'space_shuttle', 'sparkler_(fireworks)', |
| | 'spatula', 'spear', 'spectacles', 'spice_rack', 'spider', 'sponge', |
| | 'spoon', 'sportswear', 'spotlight', 'squirrel', |
| | 'stapler_(stapling_machine)', 'starfish', 'statue_(sculpture)', |
| | 'steak_(food)', 'steak_knife', 'steamer_(kitchen_appliance)', |
| | 'steering_wheel', 'stencil', 'stepladder', 'step_stool', |
| | 'stereo_(sound_system)', 'stew', 'stirrer', 'stirrup', |
| | 'stockings_(leg_wear)', 'stool', 'stop_sign', 'brake_light', 'stove', |
| | 'strainer', 'strap', 'straw_(for_drinking)', 'strawberry', |
| | 'street_sign', 'streetlight', 'string_cheese', 'stylus', 'subwoofer', |
| | 'sugar_bowl', 'sugarcane_(plant)', 'suit_(clothing)', 'sunflower', |
| | 'sunglasses', 'sunhat', 'sunscreen', 'surfboard', 'sushi', 'mop', |
| | 'sweat_pants', 'sweatband', 'sweater', 'sweatshirt', 'sweet_potato', |
| | 'swimsuit', 'sword', 'syringe', 'Tabasco_sauce', 'table-tennis_table', |
| | 'table', 'table_lamp', 'tablecloth', 'tachometer', 'taco', 'tag', |
| | 'taillight', 'tambourine', 'army_tank', 'tank_(storage_vessel)', |
| | 'tank_top_(clothing)', 'tape_(sticky_cloth_or_paper)', 'tape_measure', |
| | 'tapestry', 'tarp', 'tartan', 'tassel', 'tea_bag', 'teacup', |
| | 'teakettle', 'teapot', 'teddy_bear', 'telephone', 'telephone_booth', |
| | 'telephone_pole', 'telephoto_lens', 'television_camera', |
| | 'television_set', 'tennis_ball', 'tennis_racket', 'tequila', |
| | 'thermometer', 'thermos_bottle', 'thermostat', 'thimble', 'thread', |
| | 'thumbtack', 'tiara', 'tiger', 'tights_(clothing)', 'timer', |
| | 'tinfoil', 'tinsel', 'tissue_paper', 'toast_(food)', 'toaster', |
| | 'toaster_oven', 'toilet', 'toilet_tissue', 'tomato', 'tongs', |
| | 'toolbox', 'toothbrush', 'toothpaste', 'toothpick', 'cover', |
| | 'tortilla', 'tow_truck', 'towel', 'towel_rack', 'toy', |
| | 'tractor_(farm_equipment)', 'traffic_light', 'dirt_bike', |
| | 'trailer_truck', 'train_(railroad_vehicle)', 'trampoline', 'tray', |
| | 'tree_house', 'trench_coat', 'triangle_(musical_instrument)', |
| | 'tricycle', 'tripod', 'trousers', 'truck', 'truffle_(chocolate)', |
| | 'trunk', 'vat', 'turban', 'turkey_(bird)', 'turkey_(food)', 'turnip', |
| | 'turtle', 'turtleneck_(clothing)', 'typewriter', 'umbrella', |
| | 'underwear', 'unicycle', 'urinal', 'urn', 'vacuum_cleaner', 'valve', |
| | 'vase', 'vending_machine', 'vent', 'videotape', 'vinegar', 'violin', |
| | 'vodka', 'volleyball', 'vulture', 'waffle', 'waffle_iron', 'wagon', |
| | 'wagon_wheel', 'walking_stick', 'wall_clock', 'wall_socket', 'wallet', |
| | 'walrus', 'wardrobe', 'wasabi', 'automatic_washer', 'watch', |
| | 'water_bottle', 'water_cooler', 'water_faucet', 'water_filter', |
| | 'water_heater', 'water_jug', 'water_gun', 'water_scooter', |
| | 'water_ski', 'water_tower', 'watering_can', 'watermelon', |
| | 'weathervane', 'webcam', 'wedding_cake', 'wedding_ring', 'wet_suit', |
| | 'wheel', 'wheelchair', 'whipped_cream', 'whiskey', 'whistle', 'wick', |
| | 'wig', 'wind_chime', 'windmill', 'window_box_(for_plants)', |
| | 'windshield_wiper', 'windsock', 'wine_bottle', 'wine_bucket', |
| | 'wineglass', 'wing_chair', 'blinder_(for_horses)', 'wok', 'wolf', |
| | 'wooden_spoon', 'wreath', 'wrench', 'wristband', 'wristlet', 'yacht', |
| | 'yak', 'yogurt', 'yoke_(animal_equipment)', 'zebra', 'zucchini'), |
| | 'palette': |
| | None |
| | } |
| |
|
| | def load_data_list(self) -> List[dict]: |
| | """Load annotations from an annotation file named as ``self.ann_file`` |
| | |
| | Returns: |
| | List[dict]: A list of annotation. |
| | """ |
| | try: |
| | import lvis |
| | if getattr(lvis, '__version__', '0') >= '10.5.3': |
| | warnings.warn( |
| | 'mmlvis is deprecated, please install official lvis-api by "pip install git+https://github.com/lvis-dataset/lvis-api.git"', |
| | UserWarning) |
| | from lvis import LVIS |
| | except ImportError: |
| | raise ImportError( |
| | 'Package lvis is not installed. Please run "pip install git+https://github.com/lvis-dataset/lvis-api.git".' |
| | ) |
| | with get_local_path( |
| | self.ann_file, backend_args=self.backend_args) as local_path: |
| | self.lvis = LVIS(local_path) |
| | self.cat_ids = self.lvis.get_cat_ids() |
| | self.cat2label = {cat_id: i for i, cat_id in enumerate(self.cat_ids)} |
| | self.cat_img_map = copy.deepcopy(self.lvis.cat_img_map) |
| |
|
| | img_ids = self.lvis.get_img_ids() |
| | data_list = [] |
| | total_ann_ids = [] |
| | for img_id in img_ids: |
| | raw_img_info = self.lvis.load_imgs([img_id])[0] |
| | raw_img_info['img_id'] = img_id |
| | if raw_img_info['file_name'].startswith('COCO'): |
| | |
| | |
| | |
| | |
| | raw_img_info['file_name'] = raw_img_info['file_name'][-16:] |
| | ann_ids = self.lvis.get_ann_ids(img_ids=[img_id]) |
| | raw_ann_info = self.lvis.load_anns(ann_ids) |
| | total_ann_ids.extend(ann_ids) |
| |
|
| | parsed_data_info = self.parse_data_info({ |
| | 'raw_ann_info': |
| | raw_ann_info, |
| | 'raw_img_info': |
| | raw_img_info |
| | }) |
| | data_list.append(parsed_data_info) |
| | if self.ANN_ID_UNIQUE: |
| | assert len(set(total_ann_ids)) == len( |
| | total_ann_ids |
| | ), f"Annotation ids in '{self.ann_file}' are not unique!" |
| |
|
| | del self.lvis |
| |
|
| | return data_list |
| |
|
| |
|
| | LVISDataset = LVISV05Dataset |
| | DATASETS.register_module(name='LVISDataset', module=LVISDataset) |
| |
|
| |
|
| | @DATASETS.register_module() |
| | class LVISV1Dataset(LVISDataset): |
| | """LVIS v1 dataset for detection.""" |
| |
|
| | METAINFO = { |
| | 'classes': |
| | ('aerosol_can', 'air_conditioner', 'airplane', 'alarm_clock', |
| | 'alcohol', 'alligator', 'almond', 'ambulance', 'amplifier', 'anklet', |
| | 'antenna', 'apple', 'applesauce', 'apricot', 'apron', 'aquarium', |
| | 'arctic_(type_of_shoe)', 'armband', 'armchair', 'armoire', 'armor', |
| | 'artichoke', 'trash_can', 'ashtray', 'asparagus', 'atomizer', |
| | 'avocado', 'award', 'awning', 'ax', 'baboon', 'baby_buggy', |
| | 'basketball_backboard', 'backpack', 'handbag', 'suitcase', 'bagel', |
| | 'bagpipe', 'baguet', 'bait', 'ball', 'ballet_skirt', 'balloon', |
| | 'bamboo', 'banana', 'Band_Aid', 'bandage', 'bandanna', 'banjo', |
| | 'banner', 'barbell', 'barge', 'barrel', 'barrette', 'barrow', |
| | 'baseball_base', 'baseball', 'baseball_bat', 'baseball_cap', |
| | 'baseball_glove', 'basket', 'basketball', 'bass_horn', 'bat_(animal)', |
| | 'bath_mat', 'bath_towel', 'bathrobe', 'bathtub', 'batter_(food)', |
| | 'battery', 'beachball', 'bead', 'bean_curd', 'beanbag', 'beanie', |
| | 'bear', 'bed', 'bedpan', 'bedspread', 'cow', 'beef_(food)', 'beeper', |
| | 'beer_bottle', 'beer_can', 'beetle', 'bell', 'bell_pepper', 'belt', |
| | 'belt_buckle', 'bench', 'beret', 'bib', 'Bible', 'bicycle', 'visor', |
| | 'billboard', 'binder', 'binoculars', 'bird', 'birdfeeder', 'birdbath', |
| | 'birdcage', 'birdhouse', 'birthday_cake', 'birthday_card', |
| | 'pirate_flag', 'black_sheep', 'blackberry', 'blackboard', 'blanket', |
| | 'blazer', 'blender', 'blimp', 'blinker', 'blouse', 'blueberry', |
| | 'gameboard', 'boat', 'bob', 'bobbin', 'bobby_pin', 'boiled_egg', |
| | 'bolo_tie', 'deadbolt', 'bolt', 'bonnet', 'book', 'bookcase', |
| | 'booklet', 'bookmark', 'boom_microphone', 'boot', 'bottle', |
| | 'bottle_opener', 'bouquet', 'bow_(weapon)', |
| | 'bow_(decorative_ribbons)', 'bow-tie', 'bowl', 'pipe_bowl', |
| | 'bowler_hat', 'bowling_ball', 'box', 'boxing_glove', 'suspenders', |
| | 'bracelet', 'brass_plaque', 'brassiere', 'bread-bin', 'bread', |
| | 'breechcloth', 'bridal_gown', 'briefcase', 'broccoli', 'broach', |
| | 'broom', 'brownie', 'brussels_sprouts', 'bubble_gum', 'bucket', |
| | 'horse_buggy', 'bull', 'bulldog', 'bulldozer', 'bullet_train', |
| | 'bulletin_board', 'bulletproof_vest', 'bullhorn', 'bun', 'bunk_bed', |
| | 'buoy', 'burrito', 'bus_(vehicle)', 'business_card', 'butter', |
| | 'butterfly', 'button', 'cab_(taxi)', 'cabana', 'cabin_car', 'cabinet', |
| | 'locker', 'cake', 'calculator', 'calendar', 'calf', 'camcorder', |
| | 'camel', 'camera', 'camera_lens', 'camper_(vehicle)', 'can', |
| | 'can_opener', 'candle', 'candle_holder', 'candy_bar', 'candy_cane', |
| | 'walking_cane', 'canister', 'canoe', 'cantaloup', 'canteen', |
| | 'cap_(headwear)', 'bottle_cap', 'cape', 'cappuccino', |
| | 'car_(automobile)', 'railcar_(part_of_a_train)', 'elevator_car', |
| | 'car_battery', 'identity_card', 'card', 'cardigan', 'cargo_ship', |
| | 'carnation', 'horse_carriage', 'carrot', 'tote_bag', 'cart', 'carton', |
| | 'cash_register', 'casserole', 'cassette', 'cast', 'cat', |
| | 'cauliflower', 'cayenne_(spice)', 'CD_player', 'celery', |
| | 'cellular_telephone', 'chain_mail', 'chair', 'chaise_longue', |
| | 'chalice', 'chandelier', 'chap', 'checkbook', 'checkerboard', |
| | 'cherry', 'chessboard', 'chicken_(animal)', 'chickpea', |
| | 'chili_(vegetable)', 'chime', 'chinaware', 'crisp_(potato_chip)', |
| | 'poker_chip', 'chocolate_bar', 'chocolate_cake', 'chocolate_milk', |
| | 'chocolate_mousse', 'choker', 'chopping_board', 'chopstick', |
| | 'Christmas_tree', 'slide', 'cider', 'cigar_box', 'cigarette', |
| | 'cigarette_case', 'cistern', 'clarinet', 'clasp', 'cleansing_agent', |
| | 'cleat_(for_securing_rope)', 'clementine', 'clip', 'clipboard', |
| | 'clippers_(for_plants)', 'cloak', 'clock', 'clock_tower', |
| | 'clothes_hamper', 'clothespin', 'clutch_bag', 'coaster', 'coat', |
| | 'coat_hanger', 'coatrack', 'cock', 'cockroach', 'cocoa_(beverage)', |
| | 'coconut', 'coffee_maker', 'coffee_table', 'coffeepot', 'coil', |
| | 'coin', 'colander', 'coleslaw', 'coloring_material', |
| | 'combination_lock', 'pacifier', 'comic_book', 'compass', |
| | 'computer_keyboard', 'condiment', 'cone', 'control', |
| | 'convertible_(automobile)', 'sofa_bed', 'cooker', 'cookie', |
| | 'cooking_utensil', 'cooler_(for_food)', 'cork_(bottle_plug)', |
| | 'corkboard', 'corkscrew', 'edible_corn', 'cornbread', 'cornet', |
| | 'cornice', 'cornmeal', 'corset', 'costume', 'cougar', 'coverall', |
| | 'cowbell', 'cowboy_hat', 'crab_(animal)', 'crabmeat', 'cracker', |
| | 'crape', 'crate', 'crayon', 'cream_pitcher', 'crescent_roll', 'crib', |
| | 'crock_pot', 'crossbar', 'crouton', 'crow', 'crowbar', 'crown', |
| | 'crucifix', 'cruise_ship', 'police_cruiser', 'crumb', 'crutch', |
| | 'cub_(animal)', 'cube', 'cucumber', 'cufflink', 'cup', 'trophy_cup', |
| | 'cupboard', 'cupcake', 'hair_curler', 'curling_iron', 'curtain', |
| | 'cushion', 'cylinder', 'cymbal', 'dagger', 'dalmatian', 'dartboard', |
| | 'date_(fruit)', 'deck_chair', 'deer', 'dental_floss', 'desk', |
| | 'detergent', 'diaper', 'diary', 'die', 'dinghy', 'dining_table', |
| | 'tux', 'dish', 'dish_antenna', 'dishrag', 'dishtowel', 'dishwasher', |
| | 'dishwasher_detergent', 'dispenser', 'diving_board', 'Dixie_cup', |
| | 'dog', 'dog_collar', 'doll', 'dollar', 'dollhouse', 'dolphin', |
| | 'domestic_ass', 'doorknob', 'doormat', 'doughnut', 'dove', |
| | 'dragonfly', 'drawer', 'underdrawers', 'dress', 'dress_hat', |
| | 'dress_suit', 'dresser', 'drill', 'drone', 'dropper', |
| | 'drum_(musical_instrument)', 'drumstick', 'duck', 'duckling', |
| | 'duct_tape', 'duffel_bag', 'dumbbell', 'dumpster', 'dustpan', 'eagle', |
| | 'earphone', 'earplug', 'earring', 'easel', 'eclair', 'eel', 'egg', |
| | 'egg_roll', 'egg_yolk', 'eggbeater', 'eggplant', 'electric_chair', |
| | 'refrigerator', 'elephant', 'elk', 'envelope', 'eraser', 'escargot', |
| | 'eyepatch', 'falcon', 'fan', 'faucet', 'fedora', 'ferret', |
| | 'Ferris_wheel', 'ferry', 'fig_(fruit)', 'fighter_jet', 'figurine', |
| | 'file_cabinet', 'file_(tool)', 'fire_alarm', 'fire_engine', |
| | 'fire_extinguisher', 'fire_hose', 'fireplace', 'fireplug', |
| | 'first-aid_kit', 'fish', 'fish_(food)', 'fishbowl', 'fishing_rod', |
| | 'flag', 'flagpole', 'flamingo', 'flannel', 'flap', 'flash', |
| | 'flashlight', 'fleece', 'flip-flop_(sandal)', 'flipper_(footwear)', |
| | 'flower_arrangement', 'flute_glass', 'foal', 'folding_chair', |
| | 'food_processor', 'football_(American)', 'football_helmet', |
| | 'footstool', 'fork', 'forklift', 'freight_car', 'French_toast', |
| | 'freshener', 'frisbee', 'frog', 'fruit_juice', 'frying_pan', 'fudge', |
| | 'funnel', 'futon', 'gag', 'garbage', 'garbage_truck', 'garden_hose', |
| | 'gargle', 'gargoyle', 'garlic', 'gasmask', 'gazelle', 'gelatin', |
| | 'gemstone', 'generator', 'giant_panda', 'gift_wrap', 'ginger', |
| | 'giraffe', 'cincture', 'glass_(drink_container)', 'globe', 'glove', |
| | 'goat', 'goggles', 'goldfish', 'golf_club', 'golfcart', |
| | 'gondola_(boat)', 'goose', 'gorilla', 'gourd', 'grape', 'grater', |
| | 'gravestone', 'gravy_boat', 'green_bean', 'green_onion', 'griddle', |
| | 'grill', 'grits', 'grizzly', 'grocery_bag', 'guitar', 'gull', 'gun', |
| | 'hairbrush', 'hairnet', 'hairpin', 'halter_top', 'ham', 'hamburger', |
| | 'hammer', 'hammock', 'hamper', 'hamster', 'hair_dryer', 'hand_glass', |
| | 'hand_towel', 'handcart', 'handcuff', 'handkerchief', 'handle', |
| | 'handsaw', 'hardback_book', 'harmonium', 'hat', 'hatbox', 'veil', |
| | 'headband', 'headboard', 'headlight', 'headscarf', 'headset', |
| | 'headstall_(for_horses)', 'heart', 'heater', 'helicopter', 'helmet', |
| | 'heron', 'highchair', 'hinge', 'hippopotamus', 'hockey_stick', 'hog', |
| | 'home_plate_(baseball)', 'honey', 'fume_hood', 'hook', 'hookah', |
| | 'hornet', 'horse', 'hose', 'hot-air_balloon', 'hotplate', 'hot_sauce', |
| | 'hourglass', 'houseboat', 'hummingbird', 'hummus', 'polar_bear', |
| | 'icecream', 'popsicle', 'ice_maker', 'ice_pack', 'ice_skate', |
| | 'igniter', 'inhaler', 'iPod', 'iron_(for_clothing)', 'ironing_board', |
| | 'jacket', 'jam', 'jar', 'jean', 'jeep', 'jelly_bean', 'jersey', |
| | 'jet_plane', 'jewel', 'jewelry', 'joystick', 'jumpsuit', 'kayak', |
| | 'keg', 'kennel', 'kettle', 'key', 'keycard', 'kilt', 'kimono', |
| | 'kitchen_sink', 'kitchen_table', 'kite', 'kitten', 'kiwi_fruit', |
| | 'knee_pad', 'knife', 'knitting_needle', 'knob', 'knocker_(on_a_door)', |
| | 'koala', 'lab_coat', 'ladder', 'ladle', 'ladybug', 'lamb_(animal)', |
| | 'lamb-chop', 'lamp', 'lamppost', 'lampshade', 'lantern', 'lanyard', |
| | 'laptop_computer', 'lasagna', 'latch', 'lawn_mower', 'leather', |
| | 'legging_(clothing)', 'Lego', 'legume', 'lemon', 'lemonade', |
| | 'lettuce', 'license_plate', 'life_buoy', 'life_jacket', 'lightbulb', |
| | 'lightning_rod', 'lime', 'limousine', 'lion', 'lip_balm', 'liquor', |
| | 'lizard', 'log', 'lollipop', 'speaker_(stereo_equipment)', 'loveseat', |
| | 'machine_gun', 'magazine', 'magnet', 'mail_slot', 'mailbox_(at_home)', |
| | 'mallard', 'mallet', 'mammoth', 'manatee', 'mandarin_orange', |
| | 'manger', 'manhole', 'map', 'marker', 'martini', 'mascot', |
| | 'mashed_potato', 'masher', 'mask', 'mast', 'mat_(gym_equipment)', |
| | 'matchbox', 'mattress', 'measuring_cup', 'measuring_stick', |
| | 'meatball', 'medicine', 'melon', 'microphone', 'microscope', |
| | 'microwave_oven', 'milestone', 'milk', 'milk_can', 'milkshake', |
| | 'minivan', 'mint_candy', 'mirror', 'mitten', 'mixer_(kitchen_tool)', |
| | 'money', 'monitor_(computer_equipment) computer_monitor', 'monkey', |
| | 'motor', 'motor_scooter', 'motor_vehicle', 'motorcycle', |
| | 'mound_(baseball)', 'mouse_(computer_equipment)', 'mousepad', |
| | 'muffin', 'mug', 'mushroom', 'music_stool', 'musical_instrument', |
| | 'nailfile', 'napkin', 'neckerchief', 'necklace', 'necktie', 'needle', |
| | 'nest', 'newspaper', 'newsstand', 'nightshirt', |
| | 'nosebag_(for_animals)', 'noseband_(for_animals)', 'notebook', |
| | 'notepad', 'nut', 'nutcracker', 'oar', 'octopus_(food)', |
| | 'octopus_(animal)', 'oil_lamp', 'olive_oil', 'omelet', 'onion', |
| | 'orange_(fruit)', 'orange_juice', 'ostrich', 'ottoman', 'oven', |
| | 'overalls_(clothing)', 'owl', 'packet', 'inkpad', 'pad', 'paddle', |
| | 'padlock', 'paintbrush', 'painting', 'pajamas', 'palette', |
| | 'pan_(for_cooking)', 'pan_(metal_container)', 'pancake', 'pantyhose', |
| | 'papaya', 'paper_plate', 'paper_towel', 'paperback_book', |
| | 'paperweight', 'parachute', 'parakeet', 'parasail_(sports)', |
| | 'parasol', 'parchment', 'parka', 'parking_meter', 'parrot', |
| | 'passenger_car_(part_of_a_train)', 'passenger_ship', 'passport', |
| | 'pastry', 'patty_(food)', 'pea_(food)', 'peach', 'peanut_butter', |
| | 'pear', 'peeler_(tool_for_fruit_and_vegetables)', 'wooden_leg', |
| | 'pegboard', 'pelican', 'pen', 'pencil', 'pencil_box', |
| | 'pencil_sharpener', 'pendulum', 'penguin', 'pennant', 'penny_(coin)', |
| | 'pepper', 'pepper_mill', 'perfume', 'persimmon', 'person', 'pet', |
| | 'pew_(church_bench)', 'phonebook', 'phonograph_record', 'piano', |
| | 'pickle', 'pickup_truck', 'pie', 'pigeon', 'piggy_bank', 'pillow', |
| | 'pin_(non_jewelry)', 'pineapple', 'pinecone', 'ping-pong_ball', |
| | 'pinwheel', 'tobacco_pipe', 'pipe', 'pistol', 'pita_(bread)', |
| | 'pitcher_(vessel_for_liquid)', 'pitchfork', 'pizza', 'place_mat', |
| | 'plate', 'platter', 'playpen', 'pliers', 'plow_(farm_equipment)', |
| | 'plume', 'pocket_watch', 'pocketknife', 'poker_(fire_stirring_tool)', |
| | 'pole', 'polo_shirt', 'poncho', 'pony', 'pool_table', 'pop_(soda)', |
| | 'postbox_(public)', 'postcard', 'poster', 'pot', 'flowerpot', |
| | 'potato', 'potholder', 'pottery', 'pouch', 'power_shovel', 'prawn', |
| | 'pretzel', 'printer', 'projectile_(weapon)', 'projector', 'propeller', |
| | 'prune', 'pudding', 'puffer_(fish)', 'puffin', 'pug-dog', 'pumpkin', |
| | 'puncher', 'puppet', 'puppy', 'quesadilla', 'quiche', 'quilt', |
| | 'rabbit', 'race_car', 'racket', 'radar', 'radiator', 'radio_receiver', |
| | 'radish', 'raft', 'rag_doll', 'raincoat', 'ram_(animal)', 'raspberry', |
| | 'rat', 'razorblade', 'reamer_(juicer)', 'rearview_mirror', 'receipt', |
| | 'recliner', 'record_player', 'reflector', 'remote_control', |
| | 'rhinoceros', 'rib_(food)', 'rifle', 'ring', 'river_boat', 'road_map', |
| | 'robe', 'rocking_chair', 'rodent', 'roller_skate', 'Rollerblade', |
| | 'rolling_pin', 'root_beer', 'router_(computer_equipment)', |
| | 'rubber_band', 'runner_(carpet)', 'plastic_bag', |
| | 'saddle_(on_an_animal)', 'saddle_blanket', 'saddlebag', 'safety_pin', |
| | 'sail', 'salad', 'salad_plate', 'salami', 'salmon_(fish)', |
| | 'salmon_(food)', 'salsa', 'saltshaker', 'sandal_(type_of_shoe)', |
| | 'sandwich', 'satchel', 'saucepan', 'saucer', 'sausage', 'sawhorse', |
| | 'saxophone', 'scale_(measuring_instrument)', 'scarecrow', 'scarf', |
| | 'school_bus', 'scissors', 'scoreboard', 'scraper', 'screwdriver', |
| | 'scrubbing_brush', 'sculpture', 'seabird', 'seahorse', 'seaplane', |
| | 'seashell', 'sewing_machine', 'shaker', 'shampoo', 'shark', |
| | 'sharpener', 'Sharpie', 'shaver_(electric)', 'shaving_cream', 'shawl', |
| | 'shears', 'sheep', 'shepherd_dog', 'sherbert', 'shield', 'shirt', |
| | 'shoe', 'shopping_bag', 'shopping_cart', 'short_pants', 'shot_glass', |
| | 'shoulder_bag', 'shovel', 'shower_head', 'shower_cap', |
| | 'shower_curtain', 'shredder_(for_paper)', 'signboard', 'silo', 'sink', |
| | 'skateboard', 'skewer', 'ski', 'ski_boot', 'ski_parka', 'ski_pole', |
| | 'skirt', 'skullcap', 'sled', 'sleeping_bag', 'sling_(bandage)', |
| | 'slipper_(footwear)', 'smoothie', 'snake', 'snowboard', 'snowman', |
| | 'snowmobile', 'soap', 'soccer_ball', 'sock', 'sofa', 'softball', |
| | 'solar_array', 'sombrero', 'soup', 'soup_bowl', 'soupspoon', |
| | 'sour_cream', 'soya_milk', 'space_shuttle', 'sparkler_(fireworks)', |
| | 'spatula', 'spear', 'spectacles', 'spice_rack', 'spider', 'crawfish', |
| | 'sponge', 'spoon', 'sportswear', 'spotlight', 'squid_(food)', |
| | 'squirrel', 'stagecoach', 'stapler_(stapling_machine)', 'starfish', |
| | 'statue_(sculpture)', 'steak_(food)', 'steak_knife', 'steering_wheel', |
| | 'stepladder', 'step_stool', 'stereo_(sound_system)', 'stew', |
| | 'stirrer', 'stirrup', 'stool', 'stop_sign', 'brake_light', 'stove', |
| | 'strainer', 'strap', 'straw_(for_drinking)', 'strawberry', |
| | 'street_sign', 'streetlight', 'string_cheese', 'stylus', 'subwoofer', |
| | 'sugar_bowl', 'sugarcane_(plant)', 'suit_(clothing)', 'sunflower', |
| | 'sunglasses', 'sunhat', 'surfboard', 'sushi', 'mop', 'sweat_pants', |
| | 'sweatband', 'sweater', 'sweatshirt', 'sweet_potato', 'swimsuit', |
| | 'sword', 'syringe', 'Tabasco_sauce', 'table-tennis_table', 'table', |
| | 'table_lamp', 'tablecloth', 'tachometer', 'taco', 'tag', 'taillight', |
| | 'tambourine', 'army_tank', 'tank_(storage_vessel)', |
| | 'tank_top_(clothing)', 'tape_(sticky_cloth_or_paper)', 'tape_measure', |
| | 'tapestry', 'tarp', 'tartan', 'tassel', 'tea_bag', 'teacup', |
| | 'teakettle', 'teapot', 'teddy_bear', 'telephone', 'telephone_booth', |
| | 'telephone_pole', 'telephoto_lens', 'television_camera', |
| | 'television_set', 'tennis_ball', 'tennis_racket', 'tequila', |
| | 'thermometer', 'thermos_bottle', 'thermostat', 'thimble', 'thread', |
| | 'thumbtack', 'tiara', 'tiger', 'tights_(clothing)', 'timer', |
| | 'tinfoil', 'tinsel', 'tissue_paper', 'toast_(food)', 'toaster', |
| | 'toaster_oven', 'toilet', 'toilet_tissue', 'tomato', 'tongs', |
| | 'toolbox', 'toothbrush', 'toothpaste', 'toothpick', 'cover', |
| | 'tortilla', 'tow_truck', 'towel', 'towel_rack', 'toy', |
| | 'tractor_(farm_equipment)', 'traffic_light', 'dirt_bike', |
| | 'trailer_truck', 'train_(railroad_vehicle)', 'trampoline', 'tray', |
| | 'trench_coat', 'triangle_(musical_instrument)', 'tricycle', 'tripod', |
| | 'trousers', 'truck', 'truffle_(chocolate)', 'trunk', 'vat', 'turban', |
| | 'turkey_(food)', 'turnip', 'turtle', 'turtleneck_(clothing)', |
| | 'typewriter', 'umbrella', 'underwear', 'unicycle', 'urinal', 'urn', |
| | 'vacuum_cleaner', 'vase', 'vending_machine', 'vent', 'vest', |
| | 'videotape', 'vinegar', 'violin', 'vodka', 'volleyball', 'vulture', |
| | 'waffle', 'waffle_iron', 'wagon', 'wagon_wheel', 'walking_stick', |
| | 'wall_clock', 'wall_socket', 'wallet', 'walrus', 'wardrobe', |
| | 'washbasin', 'automatic_washer', 'watch', 'water_bottle', |
| | 'water_cooler', 'water_faucet', 'water_heater', 'water_jug', |
| | 'water_gun', 'water_scooter', 'water_ski', 'water_tower', |
| | 'watering_can', 'watermelon', 'weathervane', 'webcam', 'wedding_cake', |
| | 'wedding_ring', 'wet_suit', 'wheel', 'wheelchair', 'whipped_cream', |
| | 'whistle', 'wig', 'wind_chime', 'windmill', 'window_box_(for_plants)', |
| | 'windshield_wiper', 'windsock', 'wine_bottle', 'wine_bucket', |
| | 'wineglass', 'blinder_(for_horses)', 'wok', 'wolf', 'wooden_spoon', |
| | 'wreath', 'wrench', 'wristband', 'wristlet', 'yacht', 'yogurt', |
| | 'yoke_(animal_equipment)', 'zebra', 'zucchini'), |
| | 'palette': |
| | None |
| | } |
| |
|
| | def load_data_list(self) -> List[dict]: |
| | """Load annotations from an annotation file named as ``self.ann_file`` |
| | |
| | Returns: |
| | List[dict]: A list of annotation. |
| | """ |
| | try: |
| | import lvis |
| | if getattr(lvis, '__version__', '0') >= '10.5.3': |
| | warnings.warn( |
| | 'mmlvis is deprecated, please install official lvis-api by "pip install git+https://github.com/lvis-dataset/lvis-api.git"', |
| | UserWarning) |
| | from lvis import LVIS |
| | except ImportError: |
| | raise ImportError( |
| | 'Package lvis is not installed. Please run "pip install git+https://github.com/lvis-dataset/lvis-api.git".' |
| | ) |
| | with get_local_path( |
| | self.ann_file, backend_args=self.backend_args) as local_path: |
| | self.lvis = LVIS(local_path) |
| | self.cat_ids = self.lvis.get_cat_ids() |
| | self.cat2label = {cat_id: i for i, cat_id in enumerate(self.cat_ids)} |
| | self.cat_img_map = copy.deepcopy(self.lvis.cat_img_map) |
| |
|
| | img_ids = self.lvis.get_img_ids() |
| | data_list = [] |
| | total_ann_ids = [] |
| | for img_id in img_ids: |
| | raw_img_info = self.lvis.load_imgs([img_id])[0] |
| | raw_img_info['img_id'] = img_id |
| | |
| | |
| | |
| | raw_img_info['file_name'] = raw_img_info['coco_url'].replace( |
| | 'http://images.cocodataset.org/', '') |
| | ann_ids = self.lvis.get_ann_ids(img_ids=[img_id]) |
| | raw_ann_info = self.lvis.load_anns(ann_ids) |
| | total_ann_ids.extend(ann_ids) |
| | parsed_data_info = self.parse_data_info({ |
| | 'raw_ann_info': |
| | raw_ann_info, |
| | 'raw_img_info': |
| | raw_img_info |
| | }) |
| | data_list.append(parsed_data_info) |
| | if self.ANN_ID_UNIQUE: |
| | assert len(set(total_ann_ids)) == len( |
| | total_ann_ids |
| | ), f"Annotation ids in '{self.ann_file}' are not unique!" |
| |
|
| | del self.lvis |
| |
|
| | return data_list |
| |
|
