eliphatfs
Space.
53625b9
raw
history blame
19.6 kB
import os
import torch
feats = torch.load(os.path.join(os.path.dirname(__file__), 'lvis_cats.pt'))
categories = [
'Band_Aid',
'Bible',
'CD_player',
'Christmas_tree',
'Dixie_cup',
'Ferris_wheel',
'Lego',
'Rollerblade',
'Sharpie',
'Tabasco_sauce',
'aerosol_can',
'air_conditioner',
'airplane',
'alarm_clock',
'alcohol',
'alligator',
'almond',
'ambulance',
'amplifier',
'anklet',
'antenna',
'apple',
'apricot',
'apron',
'aquarium',
'arctic_(type_of_shoe)',
'armband',
'armchair',
'armoire',
'armor',
'army_tank',
'artichoke',
'ashtray',
'asparagus',
'atomizer',
'automatic_washer',
'avocado',
'award',
'awning',
'ax',
'baboon',
'baby_buggy',
'backpack',
'bagel',
'baguet',
'bait',
'ball',
'ballet_skirt',
'balloon',
'bamboo',
'banana',
'bandage',
'bandanna',
'banjo',
'banner',
'barbell',
'barge',
'barrel',
'barrow',
'baseball',
'baseball_bat',
'baseball_cap',
'baseball_glove',
'basket',
'basketball',
'basketball_backboard',
'bass_horn',
'bat_(animal)',
'bath_mat',
'bath_towel',
'bathrobe',
'bathtub',
'battery',
'beachball',
'bead',
'beanbag',
'beanie',
'bear',
'bed',
'bedpan',
'bedspread',
'beef_(food)',
'beeper',
'beer_bottle',
'beer_can',
'beetle',
'bell',
'bell_pepper',
'belt',
'belt_buckle',
'bench',
'beret',
'bicycle',
'billboard',
'binder',
'binoculars',
'bird',
'birdbath',
'birdcage',
'birdfeeder',
'birdhouse',
'birthday_cake',
'birthday_card',
'blackberry',
'blackboard',
'blanket',
'blazer',
'blender',
'blimp',
'blouse',
'blueberry',
'boat',
'bob',
'bobbin',
'boiled_egg',
'bolo_tie',
'bolt',
'bonnet',
'book',
'bookcase',
'booklet',
'bookmark',
'boom_microphone',
'boot',
'bottle',
'bottle_cap',
'bottle_opener',
'bouquet',
'bow-tie',
'bow_(decorative_ribbons)',
'bow_(weapon)',
'bowl',
'bowler_hat',
'bowling_ball',
'box',
'boxing_glove',
'bracelet',
'brass_plaque',
'brassiere',
'bread',
'bread-bin',
'breechcloth',
'bridal_gown',
'briefcase',
'broach',
'broccoli',
'broom',
'brownie',
'brussels_sprouts',
'bubble_gum',
'bucket',
'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',
'cake',
'calculator',
'calendar',
'calf',
'camcorder',
'camel',
'camera',
'camera_lens',
'camper_(vehicle)',
'can',
'can_opener',
'candle',
'candle_holder',
'candy_bar',
'candy_cane',
'canister',
'canoe',
'cantaloup',
'canteen',
'cap_(headwear)',
'cape',
'cappuccino',
'car_(automobile)',
'car_battery',
'card',
'cardigan',
'cargo_ship',
'carnation',
'carrot',
'cart',
'carton',
'cash_register',
'casserole',
'cassette',
'cast',
'cat',
'cauliflower',
'cayenne_(spice)',
'celery',
'cellular_telephone',
'chair',
'chaise_longue',
'chalice',
'chandelier',
'checkbook',
'checkerboard',
'cherry',
'chessboard',
'chicken_(animal)',
'chili_(vegetable)',
'chime',
'chinaware',
'chocolate_bar',
'chocolate_cake',
'chocolate_milk',
'chocolate_mousse',
'choker',
'chopping_board',
'chopstick',
'cider',
'cigar_box',
'cigarette',
'cigarette_case',
'cincture',
'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',
'coloring_material',
'combination_lock',
'comic_book',
'compass',
'computer_keyboard',
'condiment',
'cone',
'control',
'convertible_(automobile)',
'cooker',
'cookie',
'cooking_utensil',
'cooler_(for_food)',
'cork_(bottle_plug)',
'corkboard',
'corkscrew',
'cornbread',
'cornet',
'cornice',
'cornmeal',
'corset',
'costume',
'cougar',
'cover',
'coverall',
'cow',
'cowbell',
'cowboy_hat',
'crab_(animal)',
'crabmeat',
'cracker',
'crape',
'crate',
'crawfish',
'crayon',
'cream_pitcher',
'crescent_roll',
'crib',
'crisp_(potato_chip)',
'crossbar',
'crouton',
'crow',
'crowbar',
'crown',
'crucifix',
'cruise_ship',
'crutch',
'cub_(animal)',
'cube',
'cucumber',
'cufflink',
'cup',
'cupboard',
'cupcake',
'curtain',
'cushion',
'cylinder',
'cymbal',
'dagger',
'dalmatian',
'dartboard',
'date_(fruit)',
'deadbolt',
'deck_chair',
'deer',
'desk',
'detergent',
'diaper',
'diary',
'die',
'dinghy',
'dining_table',
'dirt_bike',
'dish',
'dish_antenna',
'dishrag',
'dishtowel',
'dishwasher',
'dishwasher_detergent',
'dispenser',
'dog',
'dog_collar',
'doll',
'dollar',
'dollhouse',
'dolphin',
'domestic_ass',
'doorknob',
'doormat',
'doughnut',
'dove',
'dragonfly',
'drawer',
'dress',
'dress_hat',
'dress_suit',
'dresser',
'drill',
'drone',
'drum_(musical_instrument)',
'drumstick',
'duck',
'duckling',
'duct_tape',
'duffel_bag',
'dumbbell',
'dumpster',
'dustpan',
'eagle',
'earphone',
'earplug',
'earring',
'easel',
'eclair',
'edible_corn',
'eel',
'egg',
'egg_roll',
'egg_yolk',
'eggbeater',
'eggplant',
'elephant',
'elevator_car',
'elk',
'envelope',
'eraser',
'escargot',
'eyepatch',
'falcon',
'fan',
'faucet',
'fedora',
'ferret',
'ferry',
'fig_(fruit)',
'fighter_jet',
'figurine',
'file_(tool)',
'file_cabinet',
'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',
'flowerpot',
'flute_glass',
'foal',
'folding_chair',
'food_processor',
'football_(American)',
'football_helmet',
'footstool',
'fork',
'forklift',
'freight_car',
'freshener',
'frisbee',
'frog',
'fruit_juice',
'frying_pan',
'fume_hood',
'funnel',
'futon',
'gameboard',
'garbage',
'garbage_truck',
'garden_hose',
'gargle',
'gargoyle',
'garlic',
'gasmask',
'gazelle',
'gelatin',
'gemstone',
'generator',
'giant_panda',
'gift_wrap',
'ginger',
'giraffe',
'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',
'grill',
'grits',
'grizzly',
'grocery_bag',
'guitar',
'gull',
'gun',
'hair_dryer',
'hairbrush',
'hairnet',
'halter_top',
'ham',
'hamburger',
'hammer',
'hammock',
'hamper',
'hamster',
'hand_glass',
'hand_towel',
'handbag',
'handcart',
'handcuff',
'handkerchief',
'handle',
'handsaw',
'hardback_book',
'harmonium',
'hat',
'hatbox',
'headband',
'headboard',
'headlight',
'headscarf',
'headset',
'headstall_(for_horses)',
'heart',
'heater',
'helicopter',
'helmet',
'heron',
'highchair',
'hinge',
'hippopotamus',
'hockey_stick',
'hog',
'honey',
'hook',
'hookah',
'horned_cow',
'hornet',
'horse',
'horse_buggy',
'horse_carriage',
'hose',
'hot-air_balloon',
'hot_sauce',
'hotplate',
'hourglass',
'houseboat',
'hummingbird',
'iPod',
'ice_maker',
'ice_pack',
'ice_skate',
'icecream',
'identity_card',
'igniter',
'inhaler',
'inkpad',
'iron_(for_clothing)',
'ironing_board',
'jacket',
'jam',
'jar',
'jean',
'jeep',
'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-chop',
'lamb_(animal)',
'lamp',
'lamppost',
'lampshade',
'lantern',
'laptop_computer',
'lasagna',
'latch',
'lawn_mower',
'leather',
'legging_(clothing)',
'legume',
'lemon',
'lemonade',
'lettuce',
'license_plate',
'life_buoy',
'life_jacket',
'lightbulb',
'lightning_rod',
'lime',
'limousine',
'lion',
'lip_balm',
'liquor',
'lizard',
'locker',
'log',
'lollipop',
'loveseat',
'machine_gun',
'magazine',
'magnet',
'mail_slot',
'mailbox_(at_home)',
'mallard',
'mallet',
'mammoth',
'manatee',
'mandarin_orange',
'manger',
'manhole',
'map',
'marker',
'martini',
'mascot',
'mashed_potato',
'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',
'mop',
'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',
'notebook',
'notepad',
'nut',
'nutcracker',
'oar',
'octopus_(animal)',
'octopus_(food)',
'oil_lamp',
'olive_oil',
'omelet',
'onion',
'orange_(fruit)',
'orange_juice',
'ostrich',
'ottoman',
'oven',
'overalls_(clothing)',
'owl',
'pacifier',
'packet',
'paddle',
'padlock',
'paintbrush',
'painting',
'pajamas',
'palette',
'pan_(for_cooking)',
'pan_(metal_container)',
'pancake',
'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)',
'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',
'pineapple',
'pinecone',
'ping-pong_ball',
'pinwheel',
'pipe',
'pipe_bowl',
'pirate_flag',
'pistol',
'pita_(bread)',
'pitcher_(vessel_for_liquid)',
'pitchfork',
'pizza',
'place_mat',
'plastic_bag',
'plate',
'platter',
'playpen',
'pliers',
'plow_(farm_equipment)',
'plume',
'pocket_watch',
'pocketknife',
'poker_(fire_stirring_tool)',
'poker_chip',
'polar_bear',
'pole',
'police_cruiser',
'polo_shirt',
'poncho',
'pony',
'pool_table',
'pop_(soda)',
'popsicle',
'postbox_(public)',
'postcard',
'poster',
'pot',
'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',
'railcar_(part_of_a_train)',
'raincoat',
'ram_(animal)',
'raspberry',
'rat',
'reamer_(juicer)',
'rearview_mirror',
'receipt',
'recliner',
'record_player',
'reflector',
'refrigerator',
'remote_control',
'rhinoceros',
'rib_(food)',
'rifle',
'ring',
'river_boat',
'road_map',
'robe',
'rocking_chair',
'rodent',
'roller_skate',
'rolling_pin',
'root_beer',
'router_(computer_equipment)',
'rubber_band',
'runner_(carpet)',
'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',
'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_cap',
'shower_curtain',
'shower_head',
'shredder_(for_paper)',
'signboard',
'silo',
'sink',
'skateboard',
'skewer',
'ski',
'ski_boot',
'ski_parka',
'ski_pole',
'skirt',
'skullcap',
'sled',
'sleeping_bag',
'slide',
'slipper_(footwear)',
'smoothie',
'snake',
'snowboard',
'snowman',
'snowmobile',
'soap',
'soccer_ball',
'sock',
'sofa',
'sofa_bed',
'softball',
'solar_array',
'sombrero',
'soup',
'soup_bowl',
'soupspoon',
'soya_milk',
'space_shuttle',
'sparkler_(fireworks)',
'spatula',
'speaker_(stero_equipment)',
'spear',
'spectacles',
'spice_rack',
'spider',
'sponge',
'spoon',
'sportswear',
'spotlight',
'squid_(food)',
'squirrel',
'stagecoach',
'stapler_(stapling_machine)',
'starfish',
'statue_(sculpture)',
'steak_(food)',
'steak_knife',
'steering_wheel',
'step_stool',
'stepladder',
'stereo_(sound_system)',
'stew',
'stirrer',
'stirrup',
'stool',
'stop_sign',
'stove',
'strainer',
'strap',
'straw_(for_drinking)',
'strawberry',
'street_sign',
'streetlight',
'string_cheese',
'stylus',
'subwoofer',
'sugar_bowl',
'sugarcane_(plant)',
'suit_(clothing)',
'suitcase',
'sunflower',
'sunglasses',
'sunhat',
'surfboard',
'sushi',
'suspenders',
'sweat_pants',
'sweatband',
'sweater',
'sweatshirt',
'sweet_potato',
'swimsuit',
'sword',
'syringe',
'table',
'table-tennis_table',
'table_lamp',
'tablecloth',
'tachometer',
'taco',
'tag',
'taillight',
'tambourine',
'tank_(storage_vessel)',
'tank_top_(clothing)',
'tape_(sticky_cloth_or_paper)',
'tape_measure',
'tapestry',
'tarp',
'tartan',
'tassel',
'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',
'tobacco_pipe',
'toilet',
'toilet_tissue',
'tomato',
'tongs',
'toolbox',
'toothbrush',
'toothpaste',
'toothpick',
'tortilla',
'tote_bag',
'tow_truck',
'towel',
'towel_rack',
'toy',
'tractor_(farm_equipment)',
'traffic_light',
'trailer_truck',
'train_(railroad_vehicle)',
'trampoline',
'trash_can',
'tray',
'trench_coat',
'triangle_(musical_instrument)',
'tricycle',
'tripod',
'trophy_cup',
'trousers',
'truck',
'truffle_(chocolate)',
'trunk',
'turban',
'turkey_(food)',
'turnip',
'turtle',
'turtleneck_(clothing)',
'tux',
'typewriter',
'umbrella',
'underdrawers',
'underwear',
'unicycle',
'urinal',
'urn',
'vacuum_cleaner',
'vase',
'veil',
'vending_machine',
'vent',
'vest',
'videotape',
'vinegar',
'violin',
'visor',
'vodka',
'volleyball',
'vulture',
'waffle',
'waffle_iron',
'wagon',
'walking_cane',
'walking_stick',
'wall_clock',
'wall_socket',
'wallet',
'walrus',
'wardrobe',
'washbasin',
'watch',
'water_bottle',
'water_cooler',
'water_faucet',
'water_gun',
'water_heater',
'water_jug',
'water_scooter',
'water_ski',
'water_tower',
'watering_can',
'watermelon',
'weathervane',
'webcam',
'wedding_cake',
'wedding_ring',
'wet_suit',
'wheel',
'wheelchair',
'whipped_cream',
'wig',
'wind_chime',
'windmill',
'window_box_(for_plants)',
'windsock',
'wine_bottle',
'wine_bucket',
'wineglass',
'wok',
'wolf',
'wooden_leg',
'wooden_spoon',
'wreath',
'wrench',
'wristband',
'wristlet',
'yacht',
'yogurt',
'zebra',
'zucchini'
]