1555 lines
28 KiB
Python
1555 lines
28 KiB
Python
"""
|
|
This file is part of the private API. Please do not refer to any variables defined here directly as they will be
|
|
removed on future versions without warning.
|
|
"""
|
|
|
|
# This will eventually be replaced with a call at torchvision.datasets.info("imagenet").categories
|
|
_IMAGENET_CATEGORIES = [
|
|
"tench",
|
|
"goldfish",
|
|
"great white shark",
|
|
"tiger shark",
|
|
"hammerhead",
|
|
"electric ray",
|
|
"stingray",
|
|
"cock",
|
|
"hen",
|
|
"ostrich",
|
|
"brambling",
|
|
"goldfinch",
|
|
"house finch",
|
|
"junco",
|
|
"indigo bunting",
|
|
"robin",
|
|
"bulbul",
|
|
"jay",
|
|
"magpie",
|
|
"chickadee",
|
|
"water ouzel",
|
|
"kite",
|
|
"bald eagle",
|
|
"vulture",
|
|
"great grey owl",
|
|
"European fire salamander",
|
|
"common newt",
|
|
"eft",
|
|
"spotted salamander",
|
|
"axolotl",
|
|
"bullfrog",
|
|
"tree frog",
|
|
"tailed frog",
|
|
"loggerhead",
|
|
"leatherback turtle",
|
|
"mud turtle",
|
|
"terrapin",
|
|
"box turtle",
|
|
"banded gecko",
|
|
"common iguana",
|
|
"American chameleon",
|
|
"whiptail",
|
|
"agama",
|
|
"frilled lizard",
|
|
"alligator lizard",
|
|
"Gila monster",
|
|
"green lizard",
|
|
"African chameleon",
|
|
"Komodo dragon",
|
|
"African crocodile",
|
|
"American alligator",
|
|
"triceratops",
|
|
"thunder snake",
|
|
"ringneck snake",
|
|
"hognose snake",
|
|
"green snake",
|
|
"king snake",
|
|
"garter snake",
|
|
"water snake",
|
|
"vine snake",
|
|
"night snake",
|
|
"boa constrictor",
|
|
"rock python",
|
|
"Indian cobra",
|
|
"green mamba",
|
|
"sea snake",
|
|
"horned viper",
|
|
"diamondback",
|
|
"sidewinder",
|
|
"trilobite",
|
|
"harvestman",
|
|
"scorpion",
|
|
"black and gold garden spider",
|
|
"barn spider",
|
|
"garden spider",
|
|
"black widow",
|
|
"tarantula",
|
|
"wolf spider",
|
|
"tick",
|
|
"centipede",
|
|
"black grouse",
|
|
"ptarmigan",
|
|
"ruffed grouse",
|
|
"prairie chicken",
|
|
"peacock",
|
|
"quail",
|
|
"partridge",
|
|
"African grey",
|
|
"macaw",
|
|
"sulphur-crested cockatoo",
|
|
"lorikeet",
|
|
"coucal",
|
|
"bee eater",
|
|
"hornbill",
|
|
"hummingbird",
|
|
"jacamar",
|
|
"toucan",
|
|
"drake",
|
|
"red-breasted merganser",
|
|
"goose",
|
|
"black swan",
|
|
"tusker",
|
|
"echidna",
|
|
"platypus",
|
|
"wallaby",
|
|
"koala",
|
|
"wombat",
|
|
"jellyfish",
|
|
"sea anemone",
|
|
"brain coral",
|
|
"flatworm",
|
|
"nematode",
|
|
"conch",
|
|
"snail",
|
|
"slug",
|
|
"sea slug",
|
|
"chiton",
|
|
"chambered nautilus",
|
|
"Dungeness crab",
|
|
"rock crab",
|
|
"fiddler crab",
|
|
"king crab",
|
|
"American lobster",
|
|
"spiny lobster",
|
|
"crayfish",
|
|
"hermit crab",
|
|
"isopod",
|
|
"white stork",
|
|
"black stork",
|
|
"spoonbill",
|
|
"flamingo",
|
|
"little blue heron",
|
|
"American egret",
|
|
"bittern",
|
|
"crane bird",
|
|
"limpkin",
|
|
"European gallinule",
|
|
"American coot",
|
|
"bustard",
|
|
"ruddy turnstone",
|
|
"red-backed sandpiper",
|
|
"redshank",
|
|
"dowitcher",
|
|
"oystercatcher",
|
|
"pelican",
|
|
"king penguin",
|
|
"albatross",
|
|
"grey whale",
|
|
"killer whale",
|
|
"dugong",
|
|
"sea lion",
|
|
"Chihuahua",
|
|
"Japanese spaniel",
|
|
"Maltese dog",
|
|
"Pekinese",
|
|
"Shih-Tzu",
|
|
"Blenheim spaniel",
|
|
"papillon",
|
|
"toy terrier",
|
|
"Rhodesian ridgeback",
|
|
"Afghan hound",
|
|
"basset",
|
|
"beagle",
|
|
"bloodhound",
|
|
"bluetick",
|
|
"black-and-tan coonhound",
|
|
"Walker hound",
|
|
"English foxhound",
|
|
"redbone",
|
|
"borzoi",
|
|
"Irish wolfhound",
|
|
"Italian greyhound",
|
|
"whippet",
|
|
"Ibizan hound",
|
|
"Norwegian elkhound",
|
|
"otterhound",
|
|
"Saluki",
|
|
"Scottish deerhound",
|
|
"Weimaraner",
|
|
"Staffordshire bullterrier",
|
|
"American Staffordshire terrier",
|
|
"Bedlington terrier",
|
|
"Border terrier",
|
|
"Kerry blue terrier",
|
|
"Irish terrier",
|
|
"Norfolk terrier",
|
|
"Norwich terrier",
|
|
"Yorkshire terrier",
|
|
"wire-haired fox terrier",
|
|
"Lakeland terrier",
|
|
"Sealyham terrier",
|
|
"Airedale",
|
|
"cairn",
|
|
"Australian terrier",
|
|
"Dandie Dinmont",
|
|
"Boston bull",
|
|
"miniature schnauzer",
|
|
"giant schnauzer",
|
|
"standard schnauzer",
|
|
"Scotch terrier",
|
|
"Tibetan terrier",
|
|
"silky terrier",
|
|
"soft-coated wheaten terrier",
|
|
"West Highland white terrier",
|
|
"Lhasa",
|
|
"flat-coated retriever",
|
|
"curly-coated retriever",
|
|
"golden retriever",
|
|
"Labrador retriever",
|
|
"Chesapeake Bay retriever",
|
|
"German short-haired pointer",
|
|
"vizsla",
|
|
"English setter",
|
|
"Irish setter",
|
|
"Gordon setter",
|
|
"Brittany spaniel",
|
|
"clumber",
|
|
"English springer",
|
|
"Welsh springer spaniel",
|
|
"cocker spaniel",
|
|
"Sussex spaniel",
|
|
"Irish water spaniel",
|
|
"kuvasz",
|
|
"schipperke",
|
|
"groenendael",
|
|
"malinois",
|
|
"briard",
|
|
"kelpie",
|
|
"komondor",
|
|
"Old English sheepdog",
|
|
"Shetland sheepdog",
|
|
"collie",
|
|
"Border collie",
|
|
"Bouvier des Flandres",
|
|
"Rottweiler",
|
|
"German shepherd",
|
|
"Doberman",
|
|
"miniature pinscher",
|
|
"Greater Swiss Mountain dog",
|
|
"Bernese mountain dog",
|
|
"Appenzeller",
|
|
"EntleBucher",
|
|
"boxer",
|
|
"bull mastiff",
|
|
"Tibetan mastiff",
|
|
"French bulldog",
|
|
"Great Dane",
|
|
"Saint Bernard",
|
|
"Eskimo dog",
|
|
"malamute",
|
|
"Siberian husky",
|
|
"dalmatian",
|
|
"affenpinscher",
|
|
"basenji",
|
|
"pug",
|
|
"Leonberg",
|
|
"Newfoundland",
|
|
"Great Pyrenees",
|
|
"Samoyed",
|
|
"Pomeranian",
|
|
"chow",
|
|
"keeshond",
|
|
"Brabancon griffon",
|
|
"Pembroke",
|
|
"Cardigan",
|
|
"toy poodle",
|
|
"miniature poodle",
|
|
"standard poodle",
|
|
"Mexican hairless",
|
|
"timber wolf",
|
|
"white wolf",
|
|
"red wolf",
|
|
"coyote",
|
|
"dingo",
|
|
"dhole",
|
|
"African hunting dog",
|
|
"hyena",
|
|
"red fox",
|
|
"kit fox",
|
|
"Arctic fox",
|
|
"grey fox",
|
|
"tabby",
|
|
"tiger cat",
|
|
"Persian cat",
|
|
"Siamese cat",
|
|
"Egyptian cat",
|
|
"cougar",
|
|
"lynx",
|
|
"leopard",
|
|
"snow leopard",
|
|
"jaguar",
|
|
"lion",
|
|
"tiger",
|
|
"cheetah",
|
|
"brown bear",
|
|
"American black bear",
|
|
"ice bear",
|
|
"sloth bear",
|
|
"mongoose",
|
|
"meerkat",
|
|
"tiger beetle",
|
|
"ladybug",
|
|
"ground beetle",
|
|
"long-horned beetle",
|
|
"leaf beetle",
|
|
"dung beetle",
|
|
"rhinoceros beetle",
|
|
"weevil",
|
|
"fly",
|
|
"bee",
|
|
"ant",
|
|
"grasshopper",
|
|
"cricket",
|
|
"walking stick",
|
|
"cockroach",
|
|
"mantis",
|
|
"cicada",
|
|
"leafhopper",
|
|
"lacewing",
|
|
"dragonfly",
|
|
"damselfly",
|
|
"admiral",
|
|
"ringlet",
|
|
"monarch",
|
|
"cabbage butterfly",
|
|
"sulphur butterfly",
|
|
"lycaenid",
|
|
"starfish",
|
|
"sea urchin",
|
|
"sea cucumber",
|
|
"wood rabbit",
|
|
"hare",
|
|
"Angora",
|
|
"hamster",
|
|
"porcupine",
|
|
"fox squirrel",
|
|
"marmot",
|
|
"beaver",
|
|
"guinea pig",
|
|
"sorrel",
|
|
"zebra",
|
|
"hog",
|
|
"wild boar",
|
|
"warthog",
|
|
"hippopotamus",
|
|
"ox",
|
|
"water buffalo",
|
|
"bison",
|
|
"ram",
|
|
"bighorn",
|
|
"ibex",
|
|
"hartebeest",
|
|
"impala",
|
|
"gazelle",
|
|
"Arabian camel",
|
|
"llama",
|
|
"weasel",
|
|
"mink",
|
|
"polecat",
|
|
"black-footed ferret",
|
|
"otter",
|
|
"skunk",
|
|
"badger",
|
|
"armadillo",
|
|
"three-toed sloth",
|
|
"orangutan",
|
|
"gorilla",
|
|
"chimpanzee",
|
|
"gibbon",
|
|
"siamang",
|
|
"guenon",
|
|
"patas",
|
|
"baboon",
|
|
"macaque",
|
|
"langur",
|
|
"colobus",
|
|
"proboscis monkey",
|
|
"marmoset",
|
|
"capuchin",
|
|
"howler monkey",
|
|
"titi",
|
|
"spider monkey",
|
|
"squirrel monkey",
|
|
"Madagascar cat",
|
|
"indri",
|
|
"Indian elephant",
|
|
"African elephant",
|
|
"lesser panda",
|
|
"giant panda",
|
|
"barracouta",
|
|
"eel",
|
|
"coho",
|
|
"rock beauty",
|
|
"anemone fish",
|
|
"sturgeon",
|
|
"gar",
|
|
"lionfish",
|
|
"puffer",
|
|
"abacus",
|
|
"abaya",
|
|
"academic gown",
|
|
"accordion",
|
|
"acoustic guitar",
|
|
"aircraft carrier",
|
|
"airliner",
|
|
"airship",
|
|
"altar",
|
|
"ambulance",
|
|
"amphibian",
|
|
"analog clock",
|
|
"apiary",
|
|
"apron",
|
|
"ashcan",
|
|
"assault rifle",
|
|
"backpack",
|
|
"bakery",
|
|
"balance beam",
|
|
"balloon",
|
|
"ballpoint",
|
|
"Band Aid",
|
|
"banjo",
|
|
"bannister",
|
|
"barbell",
|
|
"barber chair",
|
|
"barbershop",
|
|
"barn",
|
|
"barometer",
|
|
"barrel",
|
|
"barrow",
|
|
"baseball",
|
|
"basketball",
|
|
"bassinet",
|
|
"bassoon",
|
|
"bathing cap",
|
|
"bath towel",
|
|
"bathtub",
|
|
"beach wagon",
|
|
"beacon",
|
|
"beaker",
|
|
"bearskin",
|
|
"beer bottle",
|
|
"beer glass",
|
|
"bell cote",
|
|
"bib",
|
|
"bicycle-built-for-two",
|
|
"bikini",
|
|
"binder",
|
|
"binoculars",
|
|
"birdhouse",
|
|
"boathouse",
|
|
"bobsled",
|
|
"bolo tie",
|
|
"bonnet",
|
|
"bookcase",
|
|
"bookshop",
|
|
"bottlecap",
|
|
"bow",
|
|
"bow tie",
|
|
"brass",
|
|
"brassiere",
|
|
"breakwater",
|
|
"breastplate",
|
|
"broom",
|
|
"bucket",
|
|
"buckle",
|
|
"bulletproof vest",
|
|
"bullet train",
|
|
"butcher shop",
|
|
"cab",
|
|
"caldron",
|
|
"candle",
|
|
"cannon",
|
|
"canoe",
|
|
"can opener",
|
|
"cardigan",
|
|
"car mirror",
|
|
"carousel",
|
|
"carpenter's kit",
|
|
"carton",
|
|
"car wheel",
|
|
"cash machine",
|
|
"cassette",
|
|
"cassette player",
|
|
"castle",
|
|
"catamaran",
|
|
"CD player",
|
|
"cello",
|
|
"cellular telephone",
|
|
"chain",
|
|
"chainlink fence",
|
|
"chain mail",
|
|
"chain saw",
|
|
"chest",
|
|
"chiffonier",
|
|
"chime",
|
|
"china cabinet",
|
|
"Christmas stocking",
|
|
"church",
|
|
"cinema",
|
|
"cleaver",
|
|
"cliff dwelling",
|
|
"cloak",
|
|
"clog",
|
|
"cocktail shaker",
|
|
"coffee mug",
|
|
"coffeepot",
|
|
"coil",
|
|
"combination lock",
|
|
"computer keyboard",
|
|
"confectionery",
|
|
"container ship",
|
|
"convertible",
|
|
"corkscrew",
|
|
"cornet",
|
|
"cowboy boot",
|
|
"cowboy hat",
|
|
"cradle",
|
|
"crane",
|
|
"crash helmet",
|
|
"crate",
|
|
"crib",
|
|
"Crock Pot",
|
|
"croquet ball",
|
|
"crutch",
|
|
"cuirass",
|
|
"dam",
|
|
"desk",
|
|
"desktop computer",
|
|
"dial telephone",
|
|
"diaper",
|
|
"digital clock",
|
|
"digital watch",
|
|
"dining table",
|
|
"dishrag",
|
|
"dishwasher",
|
|
"disk brake",
|
|
"dock",
|
|
"dogsled",
|
|
"dome",
|
|
"doormat",
|
|
"drilling platform",
|
|
"drum",
|
|
"drumstick",
|
|
"dumbbell",
|
|
"Dutch oven",
|
|
"electric fan",
|
|
"electric guitar",
|
|
"electric locomotive",
|
|
"entertainment center",
|
|
"envelope",
|
|
"espresso maker",
|
|
"face powder",
|
|
"feather boa",
|
|
"file",
|
|
"fireboat",
|
|
"fire engine",
|
|
"fire screen",
|
|
"flagpole",
|
|
"flute",
|
|
"folding chair",
|
|
"football helmet",
|
|
"forklift",
|
|
"fountain",
|
|
"fountain pen",
|
|
"four-poster",
|
|
"freight car",
|
|
"French horn",
|
|
"frying pan",
|
|
"fur coat",
|
|
"garbage truck",
|
|
"gasmask",
|
|
"gas pump",
|
|
"goblet",
|
|
"go-kart",
|
|
"golf ball",
|
|
"golfcart",
|
|
"gondola",
|
|
"gong",
|
|
"gown",
|
|
"grand piano",
|
|
"greenhouse",
|
|
"grille",
|
|
"grocery store",
|
|
"guillotine",
|
|
"hair slide",
|
|
"hair spray",
|
|
"half track",
|
|
"hammer",
|
|
"hamper",
|
|
"hand blower",
|
|
"hand-held computer",
|
|
"handkerchief",
|
|
"hard disc",
|
|
"harmonica",
|
|
"harp",
|
|
"harvester",
|
|
"hatchet",
|
|
"holster",
|
|
"home theater",
|
|
"honeycomb",
|
|
"hook",
|
|
"hoopskirt",
|
|
"horizontal bar",
|
|
"horse cart",
|
|
"hourglass",
|
|
"iPod",
|
|
"iron",
|
|
"jack-o'-lantern",
|
|
"jean",
|
|
"jeep",
|
|
"jersey",
|
|
"jigsaw puzzle",
|
|
"jinrikisha",
|
|
"joystick",
|
|
"kimono",
|
|
"knee pad",
|
|
"knot",
|
|
"lab coat",
|
|
"ladle",
|
|
"lampshade",
|
|
"laptop",
|
|
"lawn mower",
|
|
"lens cap",
|
|
"letter opener",
|
|
"library",
|
|
"lifeboat",
|
|
"lighter",
|
|
"limousine",
|
|
"liner",
|
|
"lipstick",
|
|
"Loafer",
|
|
"lotion",
|
|
"loudspeaker",
|
|
"loupe",
|
|
"lumbermill",
|
|
"magnetic compass",
|
|
"mailbag",
|
|
"mailbox",
|
|
"maillot",
|
|
"maillot tank suit",
|
|
"manhole cover",
|
|
"maraca",
|
|
"marimba",
|
|
"mask",
|
|
"matchstick",
|
|
"maypole",
|
|
"maze",
|
|
"measuring cup",
|
|
"medicine chest",
|
|
"megalith",
|
|
"microphone",
|
|
"microwave",
|
|
"military uniform",
|
|
"milk can",
|
|
"minibus",
|
|
"miniskirt",
|
|
"minivan",
|
|
"missile",
|
|
"mitten",
|
|
"mixing bowl",
|
|
"mobile home",
|
|
"Model T",
|
|
"modem",
|
|
"monastery",
|
|
"monitor",
|
|
"moped",
|
|
"mortar",
|
|
"mortarboard",
|
|
"mosque",
|
|
"mosquito net",
|
|
"motor scooter",
|
|
"mountain bike",
|
|
"mountain tent",
|
|
"mouse",
|
|
"mousetrap",
|
|
"moving van",
|
|
"muzzle",
|
|
"nail",
|
|
"neck brace",
|
|
"necklace",
|
|
"nipple",
|
|
"notebook",
|
|
"obelisk",
|
|
"oboe",
|
|
"ocarina",
|
|
"odometer",
|
|
"oil filter",
|
|
"organ",
|
|
"oscilloscope",
|
|
"overskirt",
|
|
"oxcart",
|
|
"oxygen mask",
|
|
"packet",
|
|
"paddle",
|
|
"paddlewheel",
|
|
"padlock",
|
|
"paintbrush",
|
|
"pajama",
|
|
"palace",
|
|
"panpipe",
|
|
"paper towel",
|
|
"parachute",
|
|
"parallel bars",
|
|
"park bench",
|
|
"parking meter",
|
|
"passenger car",
|
|
"patio",
|
|
"pay-phone",
|
|
"pedestal",
|
|
"pencil box",
|
|
"pencil sharpener",
|
|
"perfume",
|
|
"Petri dish",
|
|
"photocopier",
|
|
"pick",
|
|
"pickelhaube",
|
|
"picket fence",
|
|
"pickup",
|
|
"pier",
|
|
"piggy bank",
|
|
"pill bottle",
|
|
"pillow",
|
|
"ping-pong ball",
|
|
"pinwheel",
|
|
"pirate",
|
|
"pitcher",
|
|
"plane",
|
|
"planetarium",
|
|
"plastic bag",
|
|
"plate rack",
|
|
"plow",
|
|
"plunger",
|
|
"Polaroid camera",
|
|
"pole",
|
|
"police van",
|
|
"poncho",
|
|
"pool table",
|
|
"pop bottle",
|
|
"pot",
|
|
"potter's wheel",
|
|
"power drill",
|
|
"prayer rug",
|
|
"printer",
|
|
"prison",
|
|
"projectile",
|
|
"projector",
|
|
"puck",
|
|
"punching bag",
|
|
"purse",
|
|
"quill",
|
|
"quilt",
|
|
"racer",
|
|
"racket",
|
|
"radiator",
|
|
"radio",
|
|
"radio telescope",
|
|
"rain barrel",
|
|
"recreational vehicle",
|
|
"reel",
|
|
"reflex camera",
|
|
"refrigerator",
|
|
"remote control",
|
|
"restaurant",
|
|
"revolver",
|
|
"rifle",
|
|
"rocking chair",
|
|
"rotisserie",
|
|
"rubber eraser",
|
|
"rugby ball",
|
|
"rule",
|
|
"running shoe",
|
|
"safe",
|
|
"safety pin",
|
|
"saltshaker",
|
|
"sandal",
|
|
"sarong",
|
|
"sax",
|
|
"scabbard",
|
|
"scale",
|
|
"school bus",
|
|
"schooner",
|
|
"scoreboard",
|
|
"screen",
|
|
"screw",
|
|
"screwdriver",
|
|
"seat belt",
|
|
"sewing machine",
|
|
"shield",
|
|
"shoe shop",
|
|
"shoji",
|
|
"shopping basket",
|
|
"shopping cart",
|
|
"shovel",
|
|
"shower cap",
|
|
"shower curtain",
|
|
"ski",
|
|
"ski mask",
|
|
"sleeping bag",
|
|
"slide rule",
|
|
"sliding door",
|
|
"slot",
|
|
"snorkel",
|
|
"snowmobile",
|
|
"snowplow",
|
|
"soap dispenser",
|
|
"soccer ball",
|
|
"sock",
|
|
"solar dish",
|
|
"sombrero",
|
|
"soup bowl",
|
|
"space bar",
|
|
"space heater",
|
|
"space shuttle",
|
|
"spatula",
|
|
"speedboat",
|
|
"spider web",
|
|
"spindle",
|
|
"sports car",
|
|
"spotlight",
|
|
"stage",
|
|
"steam locomotive",
|
|
"steel arch bridge",
|
|
"steel drum",
|
|
"stethoscope",
|
|
"stole",
|
|
"stone wall",
|
|
"stopwatch",
|
|
"stove",
|
|
"strainer",
|
|
"streetcar",
|
|
"stretcher",
|
|
"studio couch",
|
|
"stupa",
|
|
"submarine",
|
|
"suit",
|
|
"sundial",
|
|
"sunglass",
|
|
"sunglasses",
|
|
"sunscreen",
|
|
"suspension bridge",
|
|
"swab",
|
|
"sweatshirt",
|
|
"swimming trunks",
|
|
"swing",
|
|
"switch",
|
|
"syringe",
|
|
"table lamp",
|
|
"tank",
|
|
"tape player",
|
|
"teapot",
|
|
"teddy",
|
|
"television",
|
|
"tennis ball",
|
|
"thatch",
|
|
"theater curtain",
|
|
"thimble",
|
|
"thresher",
|
|
"throne",
|
|
"tile roof",
|
|
"toaster",
|
|
"tobacco shop",
|
|
"toilet seat",
|
|
"torch",
|
|
"totem pole",
|
|
"tow truck",
|
|
"toyshop",
|
|
"tractor",
|
|
"trailer truck",
|
|
"tray",
|
|
"trench coat",
|
|
"tricycle",
|
|
"trimaran",
|
|
"tripod",
|
|
"triumphal arch",
|
|
"trolleybus",
|
|
"trombone",
|
|
"tub",
|
|
"turnstile",
|
|
"typewriter keyboard",
|
|
"umbrella",
|
|
"unicycle",
|
|
"upright",
|
|
"vacuum",
|
|
"vase",
|
|
"vault",
|
|
"velvet",
|
|
"vending machine",
|
|
"vestment",
|
|
"viaduct",
|
|
"violin",
|
|
"volleyball",
|
|
"waffle iron",
|
|
"wall clock",
|
|
"wallet",
|
|
"wardrobe",
|
|
"warplane",
|
|
"washbasin",
|
|
"washer",
|
|
"water bottle",
|
|
"water jug",
|
|
"water tower",
|
|
"whiskey jug",
|
|
"whistle",
|
|
"wig",
|
|
"window screen",
|
|
"window shade",
|
|
"Windsor tie",
|
|
"wine bottle",
|
|
"wing",
|
|
"wok",
|
|
"wooden spoon",
|
|
"wool",
|
|
"worm fence",
|
|
"wreck",
|
|
"yawl",
|
|
"yurt",
|
|
"web site",
|
|
"comic book",
|
|
"crossword puzzle",
|
|
"street sign",
|
|
"traffic light",
|
|
"book jacket",
|
|
"menu",
|
|
"plate",
|
|
"guacamole",
|
|
"consomme",
|
|
"hot pot",
|
|
"trifle",
|
|
"ice cream",
|
|
"ice lolly",
|
|
"French loaf",
|
|
"bagel",
|
|
"pretzel",
|
|
"cheeseburger",
|
|
"hotdog",
|
|
"mashed potato",
|
|
"head cabbage",
|
|
"broccoli",
|
|
"cauliflower",
|
|
"zucchini",
|
|
"spaghetti squash",
|
|
"acorn squash",
|
|
"butternut squash",
|
|
"cucumber",
|
|
"artichoke",
|
|
"bell pepper",
|
|
"cardoon",
|
|
"mushroom",
|
|
"Granny Smith",
|
|
"strawberry",
|
|
"orange",
|
|
"lemon",
|
|
"fig",
|
|
"pineapple",
|
|
"banana",
|
|
"jackfruit",
|
|
"custard apple",
|
|
"pomegranate",
|
|
"hay",
|
|
"carbonara",
|
|
"chocolate sauce",
|
|
"dough",
|
|
"meat loaf",
|
|
"pizza",
|
|
"potpie",
|
|
"burrito",
|
|
"red wine",
|
|
"espresso",
|
|
"cup",
|
|
"eggnog",
|
|
"alp",
|
|
"bubble",
|
|
"cliff",
|
|
"coral reef",
|
|
"geyser",
|
|
"lakeside",
|
|
"promontory",
|
|
"sandbar",
|
|
"seashore",
|
|
"valley",
|
|
"volcano",
|
|
"ballplayer",
|
|
"groom",
|
|
"scuba diver",
|
|
"rapeseed",
|
|
"daisy",
|
|
"yellow lady's slipper",
|
|
"corn",
|
|
"acorn",
|
|
"hip",
|
|
"buckeye",
|
|
"coral fungus",
|
|
"agaric",
|
|
"gyromitra",
|
|
"stinkhorn",
|
|
"earthstar",
|
|
"hen-of-the-woods",
|
|
"bolete",
|
|
"ear",
|
|
"toilet tissue",
|
|
]
|
|
|
|
# To be replaced with torchvision.datasets.info("coco").categories
|
|
_COCO_CATEGORIES = [
|
|
"__background__",
|
|
"person",
|
|
"bicycle",
|
|
"car",
|
|
"motorcycle",
|
|
"airplane",
|
|
"bus",
|
|
"train",
|
|
"truck",
|
|
"boat",
|
|
"traffic light",
|
|
"fire hydrant",
|
|
"N/A",
|
|
"stop sign",
|
|
"parking meter",
|
|
"bench",
|
|
"bird",
|
|
"cat",
|
|
"dog",
|
|
"horse",
|
|
"sheep",
|
|
"cow",
|
|
"elephant",
|
|
"bear",
|
|
"zebra",
|
|
"giraffe",
|
|
"N/A",
|
|
"backpack",
|
|
"umbrella",
|
|
"N/A",
|
|
"N/A",
|
|
"handbag",
|
|
"tie",
|
|
"suitcase",
|
|
"frisbee",
|
|
"skis",
|
|
"snowboard",
|
|
"sports ball",
|
|
"kite",
|
|
"baseball bat",
|
|
"baseball glove",
|
|
"skateboard",
|
|
"surfboard",
|
|
"tennis racket",
|
|
"bottle",
|
|
"N/A",
|
|
"wine glass",
|
|
"cup",
|
|
"fork",
|
|
"knife",
|
|
"spoon",
|
|
"bowl",
|
|
"banana",
|
|
"apple",
|
|
"sandwich",
|
|
"orange",
|
|
"broccoli",
|
|
"carrot",
|
|
"hot dog",
|
|
"pizza",
|
|
"donut",
|
|
"cake",
|
|
"chair",
|
|
"couch",
|
|
"potted plant",
|
|
"bed",
|
|
"N/A",
|
|
"dining table",
|
|
"N/A",
|
|
"N/A",
|
|
"toilet",
|
|
"N/A",
|
|
"tv",
|
|
"laptop",
|
|
"mouse",
|
|
"remote",
|
|
"keyboard",
|
|
"cell phone",
|
|
"microwave",
|
|
"oven",
|
|
"toaster",
|
|
"sink",
|
|
"refrigerator",
|
|
"N/A",
|
|
"book",
|
|
"clock",
|
|
"vase",
|
|
"scissors",
|
|
"teddy bear",
|
|
"hair drier",
|
|
"toothbrush",
|
|
]
|
|
|
|
# To be replaced with torchvision.datasets.info("coco_kp")
|
|
_COCO_PERSON_CATEGORIES = ["no person", "person"]
|
|
_COCO_PERSON_KEYPOINT_NAMES = [
|
|
"nose",
|
|
"left_eye",
|
|
"right_eye",
|
|
"left_ear",
|
|
"right_ear",
|
|
"left_shoulder",
|
|
"right_shoulder",
|
|
"left_elbow",
|
|
"right_elbow",
|
|
"left_wrist",
|
|
"right_wrist",
|
|
"left_hip",
|
|
"right_hip",
|
|
"left_knee",
|
|
"right_knee",
|
|
"left_ankle",
|
|
"right_ankle",
|
|
]
|
|
|
|
# To be replaced with torchvision.datasets.info("voc").categories
|
|
_VOC_CATEGORIES = [
|
|
"__background__",
|
|
"aeroplane",
|
|
"bicycle",
|
|
"bird",
|
|
"boat",
|
|
"bottle",
|
|
"bus",
|
|
"car",
|
|
"cat",
|
|
"chair",
|
|
"cow",
|
|
"diningtable",
|
|
"dog",
|
|
"horse",
|
|
"motorbike",
|
|
"person",
|
|
"pottedplant",
|
|
"sheep",
|
|
"sofa",
|
|
"train",
|
|
"tvmonitor",
|
|
]
|
|
|
|
# To be replaced with torchvision.datasets.info("kinetics400").categories
|
|
_KINETICS400_CATEGORIES = [
|
|
"abseiling",
|
|
"air drumming",
|
|
"answering questions",
|
|
"applauding",
|
|
"applying cream",
|
|
"archery",
|
|
"arm wrestling",
|
|
"arranging flowers",
|
|
"assembling computer",
|
|
"auctioning",
|
|
"baby waking up",
|
|
"baking cookies",
|
|
"balloon blowing",
|
|
"bandaging",
|
|
"barbequing",
|
|
"bartending",
|
|
"beatboxing",
|
|
"bee keeping",
|
|
"belly dancing",
|
|
"bench pressing",
|
|
"bending back",
|
|
"bending metal",
|
|
"biking through snow",
|
|
"blasting sand",
|
|
"blowing glass",
|
|
"blowing leaves",
|
|
"blowing nose",
|
|
"blowing out candles",
|
|
"bobsledding",
|
|
"bookbinding",
|
|
"bouncing on trampoline",
|
|
"bowling",
|
|
"braiding hair",
|
|
"breading or breadcrumbing",
|
|
"breakdancing",
|
|
"brush painting",
|
|
"brushing hair",
|
|
"brushing teeth",
|
|
"building cabinet",
|
|
"building shed",
|
|
"bungee jumping",
|
|
"busking",
|
|
"canoeing or kayaking",
|
|
"capoeira",
|
|
"carrying baby",
|
|
"cartwheeling",
|
|
"carving pumpkin",
|
|
"catching fish",
|
|
"catching or throwing baseball",
|
|
"catching or throwing frisbee",
|
|
"catching or throwing softball",
|
|
"celebrating",
|
|
"changing oil",
|
|
"changing wheel",
|
|
"checking tires",
|
|
"cheerleading",
|
|
"chopping wood",
|
|
"clapping",
|
|
"clay pottery making",
|
|
"clean and jerk",
|
|
"cleaning floor",
|
|
"cleaning gutters",
|
|
"cleaning pool",
|
|
"cleaning shoes",
|
|
"cleaning toilet",
|
|
"cleaning windows",
|
|
"climbing a rope",
|
|
"climbing ladder",
|
|
"climbing tree",
|
|
"contact juggling",
|
|
"cooking chicken",
|
|
"cooking egg",
|
|
"cooking on campfire",
|
|
"cooking sausages",
|
|
"counting money",
|
|
"country line dancing",
|
|
"cracking neck",
|
|
"crawling baby",
|
|
"crossing river",
|
|
"crying",
|
|
"curling hair",
|
|
"cutting nails",
|
|
"cutting pineapple",
|
|
"cutting watermelon",
|
|
"dancing ballet",
|
|
"dancing charleston",
|
|
"dancing gangnam style",
|
|
"dancing macarena",
|
|
"deadlifting",
|
|
"decorating the christmas tree",
|
|
"digging",
|
|
"dining",
|
|
"disc golfing",
|
|
"diving cliff",
|
|
"dodgeball",
|
|
"doing aerobics",
|
|
"doing laundry",
|
|
"doing nails",
|
|
"drawing",
|
|
"dribbling basketball",
|
|
"drinking",
|
|
"drinking beer",
|
|
"drinking shots",
|
|
"driving car",
|
|
"driving tractor",
|
|
"drop kicking",
|
|
"drumming fingers",
|
|
"dunking basketball",
|
|
"dying hair",
|
|
"eating burger",
|
|
"eating cake",
|
|
"eating carrots",
|
|
"eating chips",
|
|
"eating doughnuts",
|
|
"eating hotdog",
|
|
"eating ice cream",
|
|
"eating spaghetti",
|
|
"eating watermelon",
|
|
"egg hunting",
|
|
"exercising arm",
|
|
"exercising with an exercise ball",
|
|
"extinguishing fire",
|
|
"faceplanting",
|
|
"feeding birds",
|
|
"feeding fish",
|
|
"feeding goats",
|
|
"filling eyebrows",
|
|
"finger snapping",
|
|
"fixing hair",
|
|
"flipping pancake",
|
|
"flying kite",
|
|
"folding clothes",
|
|
"folding napkins",
|
|
"folding paper",
|
|
"front raises",
|
|
"frying vegetables",
|
|
"garbage collecting",
|
|
"gargling",
|
|
"getting a haircut",
|
|
"getting a tattoo",
|
|
"giving or receiving award",
|
|
"golf chipping",
|
|
"golf driving",
|
|
"golf putting",
|
|
"grinding meat",
|
|
"grooming dog",
|
|
"grooming horse",
|
|
"gymnastics tumbling",
|
|
"hammer throw",
|
|
"headbanging",
|
|
"headbutting",
|
|
"high jump",
|
|
"high kick",
|
|
"hitting baseball",
|
|
"hockey stop",
|
|
"holding snake",
|
|
"hopscotch",
|
|
"hoverboarding",
|
|
"hugging",
|
|
"hula hooping",
|
|
"hurdling",
|
|
"hurling (sport)",
|
|
"ice climbing",
|
|
"ice fishing",
|
|
"ice skating",
|
|
"ironing",
|
|
"javelin throw",
|
|
"jetskiing",
|
|
"jogging",
|
|
"juggling balls",
|
|
"juggling fire",
|
|
"juggling soccer ball",
|
|
"jumping into pool",
|
|
"jumpstyle dancing",
|
|
"kicking field goal",
|
|
"kicking soccer ball",
|
|
"kissing",
|
|
"kitesurfing",
|
|
"knitting",
|
|
"krumping",
|
|
"laughing",
|
|
"laying bricks",
|
|
"long jump",
|
|
"lunge",
|
|
"making a cake",
|
|
"making a sandwich",
|
|
"making bed",
|
|
"making jewelry",
|
|
"making pizza",
|
|
"making snowman",
|
|
"making sushi",
|
|
"making tea",
|
|
"marching",
|
|
"massaging back",
|
|
"massaging feet",
|
|
"massaging legs",
|
|
"massaging person's head",
|
|
"milking cow",
|
|
"mopping floor",
|
|
"motorcycling",
|
|
"moving furniture",
|
|
"mowing lawn",
|
|
"news anchoring",
|
|
"opening bottle",
|
|
"opening present",
|
|
"paragliding",
|
|
"parasailing",
|
|
"parkour",
|
|
"passing American football (in game)",
|
|
"passing American football (not in game)",
|
|
"peeling apples",
|
|
"peeling potatoes",
|
|
"petting animal (not cat)",
|
|
"petting cat",
|
|
"picking fruit",
|
|
"planting trees",
|
|
"plastering",
|
|
"playing accordion",
|
|
"playing badminton",
|
|
"playing bagpipes",
|
|
"playing basketball",
|
|
"playing bass guitar",
|
|
"playing cards",
|
|
"playing cello",
|
|
"playing chess",
|
|
"playing clarinet",
|
|
"playing controller",
|
|
"playing cricket",
|
|
"playing cymbals",
|
|
"playing didgeridoo",
|
|
"playing drums",
|
|
"playing flute",
|
|
"playing guitar",
|
|
"playing harmonica",
|
|
"playing harp",
|
|
"playing ice hockey",
|
|
"playing keyboard",
|
|
"playing kickball",
|
|
"playing monopoly",
|
|
"playing organ",
|
|
"playing paintball",
|
|
"playing piano",
|
|
"playing poker",
|
|
"playing recorder",
|
|
"playing saxophone",
|
|
"playing squash or racquetball",
|
|
"playing tennis",
|
|
"playing trombone",
|
|
"playing trumpet",
|
|
"playing ukulele",
|
|
"playing violin",
|
|
"playing volleyball",
|
|
"playing xylophone",
|
|
"pole vault",
|
|
"presenting weather forecast",
|
|
"pull ups",
|
|
"pumping fist",
|
|
"pumping gas",
|
|
"punching bag",
|
|
"punching person (boxing)",
|
|
"push up",
|
|
"pushing car",
|
|
"pushing cart",
|
|
"pushing wheelchair",
|
|
"reading book",
|
|
"reading newspaper",
|
|
"recording music",
|
|
"riding a bike",
|
|
"riding camel",
|
|
"riding elephant",
|
|
"riding mechanical bull",
|
|
"riding mountain bike",
|
|
"riding mule",
|
|
"riding or walking with horse",
|
|
"riding scooter",
|
|
"riding unicycle",
|
|
"ripping paper",
|
|
"robot dancing",
|
|
"rock climbing",
|
|
"rock scissors paper",
|
|
"roller skating",
|
|
"running on treadmill",
|
|
"sailing",
|
|
"salsa dancing",
|
|
"sanding floor",
|
|
"scrambling eggs",
|
|
"scuba diving",
|
|
"setting table",
|
|
"shaking hands",
|
|
"shaking head",
|
|
"sharpening knives",
|
|
"sharpening pencil",
|
|
"shaving head",
|
|
"shaving legs",
|
|
"shearing sheep",
|
|
"shining shoes",
|
|
"shooting basketball",
|
|
"shooting goal (soccer)",
|
|
"shot put",
|
|
"shoveling snow",
|
|
"shredding paper",
|
|
"shuffling cards",
|
|
"side kick",
|
|
"sign language interpreting",
|
|
"singing",
|
|
"situp",
|
|
"skateboarding",
|
|
"ski jumping",
|
|
"skiing (not slalom or crosscountry)",
|
|
"skiing crosscountry",
|
|
"skiing slalom",
|
|
"skipping rope",
|
|
"skydiving",
|
|
"slacklining",
|
|
"slapping",
|
|
"sled dog racing",
|
|
"smoking",
|
|
"smoking hookah",
|
|
"snatch weight lifting",
|
|
"sneezing",
|
|
"sniffing",
|
|
"snorkeling",
|
|
"snowboarding",
|
|
"snowkiting",
|
|
"snowmobiling",
|
|
"somersaulting",
|
|
"spinning poi",
|
|
"spray painting",
|
|
"spraying",
|
|
"springboard diving",
|
|
"squat",
|
|
"sticking tongue out",
|
|
"stomping grapes",
|
|
"stretching arm",
|
|
"stretching leg",
|
|
"strumming guitar",
|
|
"surfing crowd",
|
|
"surfing water",
|
|
"sweeping floor",
|
|
"swimming backstroke",
|
|
"swimming breast stroke",
|
|
"swimming butterfly stroke",
|
|
"swing dancing",
|
|
"swinging legs",
|
|
"swinging on something",
|
|
"sword fighting",
|
|
"tai chi",
|
|
"taking a shower",
|
|
"tango dancing",
|
|
"tap dancing",
|
|
"tapping guitar",
|
|
"tapping pen",
|
|
"tasting beer",
|
|
"tasting food",
|
|
"testifying",
|
|
"texting",
|
|
"throwing axe",
|
|
"throwing ball",
|
|
"throwing discus",
|
|
"tickling",
|
|
"tobogganing",
|
|
"tossing coin",
|
|
"tossing salad",
|
|
"training dog",
|
|
"trapezing",
|
|
"trimming or shaving beard",
|
|
"trimming trees",
|
|
"triple jump",
|
|
"tying bow tie",
|
|
"tying knot (not on a tie)",
|
|
"tying tie",
|
|
"unboxing",
|
|
"unloading truck",
|
|
"using computer",
|
|
"using remote controller (not gaming)",
|
|
"using segway",
|
|
"vault",
|
|
"waiting in line",
|
|
"walking the dog",
|
|
"washing dishes",
|
|
"washing feet",
|
|
"washing hair",
|
|
"washing hands",
|
|
"water skiing",
|
|
"water sliding",
|
|
"watering plants",
|
|
"waxing back",
|
|
"waxing chest",
|
|
"waxing eyebrows",
|
|
"waxing legs",
|
|
"weaving basket",
|
|
"welding",
|
|
"whistling",
|
|
"windsurfing",
|
|
"wrapping present",
|
|
"wrestling",
|
|
"writing",
|
|
"yawning",
|
|
"yoga",
|
|
"zumba",
|
|
]
|