123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554 |
- """
- 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",
- ]
|