Push model using huggingface_hub.

README.md

@@ -2,81 +2,8 @@
 tags:
 - model_hub_mixin
 - pytorch_model_hub_mixin
-license: mit
-base_model:
-- microsoft/swin-base-patch4-window7-224
-pipeline_tag: image-feature-extraction
 ---
 
-This is fashion image feature extractor model.
-
-You can find object-detection model -> [https://huggingface.co/yainage90/fashion-object-detection](https://huggingface.co/yainage90/fashion-object-detection)
-You can find details of model in this github repo -> [fashion-visual-search](https://github.com/yainage90/fashion-visual-search)
-
-# 1. Model Architecture
-
-I used [microsoft/swin-base-patch4-window7-224](https://huggingface.co/microsoft/swin-base-patch4-window7-224) for base image encoder model. Just added a 128 size fully connected layer to lower embedding size.
-The dataset used anchor (product areas detected from posts) - positive (product thumbnail) image pairs. Within each batch, all samples except one's own positive were used as negative samples, training to minimize the distance between anchor-positive pairs while maximizing the distance between anchor-negative pairs. This method is known as contrastive learning, which is the training method used by OpenAI's CLIP model.
-Initially, anchor - positive - negative pairs were explicitly constructed in a 1:1:1 ratio using triplet loss, but training with in-batch negative sampling and contrastive loss showed much better performance as it allowed learning from more negative samples.
-
-<img src="image_encoder.png" width="500" alt="image_encoder">
-
-<img src="contrastive_learning.png" width="500" alt="contrastive_learning">
-
-
-# 2. Training dataset
-
-User posting images from onthelook and kream were crawled and preprocessed. First, raw data of image-product thumbnail combinations from posts were collected. Then, object detection was performed on posting images, and category classification was performed on product thumbnails to pair images of the same category together. For thumbnail category classification, a trained category classifier was used. Finally, about 290,000 anchor-positive image pairs were created for 6 categories: tops, bottoms, outer, shoes, bags, and hats.
-Finally, approximately 290,000 anchor-positive image pairs were created for 6 categories: tops, bottoms, outer, shoes, bags, and hats.
-
-<img src="data_sample.png" width="300" alt="data sample">
-
-
-# 3. Usage
-
-```python
-from PIL import Image
-import torch
-import torch.nn as nn
-import torch.nn.functional as F
-import torchvision.transforms as v2
-from transformers import AutoImageProcessor, SwinModel, SwinConfig
-from huggingface_hub import PyTorchModelHubMixin
-
-device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
-
-ckpt = "yainage90/fashion-image-feature-extractor"
-encoder_config = SwinConfig.from_pretrained(ckpt)
-encoder_image_processor = AutoImageProcessor.from_pretrained(ckpt)
-
-class ImageEncoder(nn.Module, PyTorchModelHubMixin):
-    def __init__(self):
-        super(ImageEncoder, self).__init__()
-        self.swin = SwinModel(config=encoder_config)
-        self.embedding_layer = nn.Linear(encoder_config.hidden_size, 128)
-
-    def forward(self, image_tensor):
-        features = self.swin(image_tensor).pooler_output
-        embeddings = self.embedding_layer(features)
-        embeddings = F.normalize(embeddings, p=2, dim=1)
-
-        return embeddings
-
-encoder = ImageEncoder().from_pretrained('yainage90/fashion-image-feature-extractor').to(device)
-
-transform = v2.Compose([
-    v2.Resize((encoder_config.image_size, encoder_config.image_size)),
-    v2.ToTensor(),
-    v2.Normalize(mean=encoder_image_processor.image_mean, std=encoder_image_processor.image_std),
-])
-
-image = Image.open('<path/to/image>').convert('RGB')
-image = transform(image)
-with torch.no_grad():
-    embedding = encoder(image.unsqueeze(0).to(device)).cpu().numpy()
-```
-
-<img src="detection_image1.png" width="500" alt="detection_image1">
-<img src="result_image1.png" width="700" alt="result_image1">
-<img src="detection_image2.png" width="500" alt="detection_image2">
-<img src="result_image2.png" width="700" alt="result_image2">
+This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
+- Library: [More Information Needed]
+- Docs: [More Information Needed]

config.json

@@ -19,104 +19,7 @@
   "id2label": {
     "0": "tench, Tinca tinca",
     "1": "goldfish, Carassius auratus",
-    "2": "great white shark, white shark, man-eater, man-eating shark, Carcharodon carcharias",
-    "3": "tiger shark, Galeocerdo cuvieri",
-    "4": "hammerhead, hammerhead shark",
-    "5": "electric ray, crampfish, numbfish, torpedo",
-    "6": "stingray",
-    "7": "cock",
-    "8": "hen",
-    "9": "ostrich, Struthio camelus",
     "10": "brambling, Fringilla montifringilla",
-    "11": "goldfinch, Carduelis carduelis",
-    "12": "house finch, linnet, Carpodacus mexicanus",
-    "13": "junco, snowbird",
-    "14": "indigo bunting, indigo finch, indigo bird, Passerina cyanea",
-    "15": "robin, American robin, Turdus migratorius",
-    "16": "bulbul",
-    "17": "jay",
-    "18": "magpie",
-    "19": "chickadee",
-    "20": "water ouzel, dipper",
-    "21": "kite",
-    "22": "bald eagle, American eagle, Haliaeetus leucocephalus",
-    "23": "vulture",
-    "24": "great grey owl, great gray owl, Strix nebulosa",
-    "25": "European fire salamander, Salamandra salamandra",
-    "26": "common newt, Triturus vulgaris",
-    "27": "eft",
-    "28": "spotted salamander, Ambystoma maculatum",
-    "29": "axolotl, mud puppy, Ambystoma mexicanum",
-    "30": "bullfrog, Rana catesbeiana",
-    "31": "tree frog, tree-frog",
-    "32": "tailed frog, bell toad, ribbed toad, tailed toad, Ascaphus trui",
-    "33": "loggerhead, loggerhead turtle, Caretta caretta",
-    "34": "leatherback turtle, leatherback, leathery turtle, Dermochelys coriacea",
-    "35": "mud turtle",
-    "36": "terrapin",
-    "37": "box turtle, box tortoise",
-    "38": "banded gecko",
-    "39": "common iguana, iguana, Iguana iguana",
-    "40": "American chameleon, anole, Anolis carolinensis",
-    "41": "whiptail, whiptail lizard",
-    "42": "agama",
-    "43": "frilled lizard, Chlamydosaurus kingi",
-    "44": "alligator lizard",
-    "45": "Gila monster, Heloderma suspectum",
-    "46": "green lizard, Lacerta viridis",
-    "47": "African chameleon, Chamaeleo chamaeleon",
-    "48": "Komodo dragon, Komodo lizard, dragon lizard, giant lizard, Varanus komodoensis",
-    "49": "African crocodile, Nile crocodile, Crocodylus niloticus",
-    "50": "American alligator, Alligator mississipiensis",
-    "51": "triceratops",
-    "52": "thunder snake, worm snake, Carphophis amoenus",
-    "53": "ringneck snake, ring-necked snake, ring snake",
-    "54": "hognose snake, puff adder, sand viper",
-    "55": "green snake, grass snake",
-    "56": "king snake, kingsnake",
-    "57": "garter snake, grass snake",
-    "58": "water snake",
-    "59": "vine snake",
-    "60": "night snake, Hypsiglena torquata",
-    "61": "boa constrictor, Constrictor constrictor",
-    "62": "rock python, rock snake, Python sebae",
-    "63": "Indian cobra, Naja naja",
-    "64": "green mamba",
-    "65": "sea snake",
-    "66": "horned viper, cerastes, sand viper, horned asp, Cerastes cornutus",
-    "67": "diamondback, diamondback rattlesnake, Crotalus adamanteus",
-    "68": "sidewinder, horned rattlesnake, Crotalus cerastes",
-    "69": "trilobite",
-    "70": "harvestman, daddy longlegs, Phalangium opilio",
-    "71": "scorpion",
-    "72": "black and gold garden spider, Argiope aurantia",
-    "73": "barn spider, Araneus cavaticus",
-    "74": "garden spider, Aranea diademata",
-    "75": "black widow, Latrodectus mactans",
-    "76": "tarantula",
-    "77": "wolf spider, hunting spider",
-    "78": "tick",
-    "79": "centipede",
-    "80": "black grouse",
-    "81": "ptarmigan",
-    "82": "ruffed grouse, partridge, Bonasa umbellus",
-    "83": "prairie chicken, prairie grouse, prairie fowl",
-    "84": "peacock",
-    "85": "quail",
-    "86": "partridge",
-    "87": "African grey, African gray, Psittacus erithacus",
-    "88": "macaw",
-    "89": "sulphur-crested cockatoo, Kakatoe galerita, Cacatua galerita",
-    "90": "lorikeet",
-    "91": "coucal",
-    "92": "bee eater",
-    "93": "hornbill",
-    "94": "hummingbird",
-    "95": "jacamar",
-    "96": "toucan",
-    "97": "drake",
-    "98": "red-breasted merganser, Mergus serrator",
-    "99": "goose",
     "100": "black swan, Cygnus atratus",
     "101": "tusker",
     "102": "echidna, spiny anteater, anteater",
@@ -127,6 +30,7 @@
     "107": "jellyfish",
     "108": "sea anemone, anemone",
     "109": "brain coral",
+    "11": "goldfinch, Carduelis carduelis",
     "110": "flatworm, platyhelminth",
     "111": "nematode, nematode worm, roundworm",
     "112": "conch",
@@ -137,6 +41,7 @@
     "117": "chambered nautilus, pearly nautilus, nautilus",
     "118": "Dungeness crab, Cancer magister",
     "119": "rock crab, Cancer irroratus",
+    "12": "house finch, linnet, Carpodacus mexicanus",
     "120": "fiddler crab",
     "121": "king crab, Alaska crab, Alaskan king crab, Alaska king crab, Paralithodes camtschatica",
     "122": "American lobster, Northern lobster, Maine lobster, Homarus americanus",
@@ -147,6 +52,7 @@
     "127": "white stork, Ciconia ciconia",
     "128": "black stork, Ciconia nigra",
     "129": "spoonbill",
+    "13": "junco, snowbird",
     "130": "flamingo",
     "131": "little blue heron, Egretta caerulea",
     "132": "American egret, great white heron, Egretta albus",
@@ -157,6 +63,7 @@
     "137": "American coot, marsh hen, mud hen, water hen, Fulica americana",
     "138": "bustard",
     "139": "ruddy turnstone, Arenaria interpres",
+    "14": "indigo bunting, indigo finch, indigo bird, Passerina cyanea",
     "140": "red-backed sandpiper, dunlin, Erolia alpina",
     "141": "redshank, Tringa totanus",
     "142": "dowitcher",
@@ -167,6 +74,7 @@
     "147": "grey whale, gray whale, devilfish, Eschrichtius gibbosus, Eschrichtius robustus",
     "148": "killer whale, killer, orca, grampus, sea wolf, Orcinus orca",
     "149": "dugong, Dugong dugon",
+    "15": "robin, American robin, Turdus migratorius",
     "150": "sea lion",
     "151": "Chihuahua",
     "152": "Japanese spaniel",
@@ -177,6 +85,7 @@
     "157": "papillon",
     "158": "toy terrier",
     "159": "Rhodesian ridgeback",
+    "16": "bulbul",
     "160": "Afghan hound, Afghan",
     "161": "basset, basset hound",
     "162": "beagle",
@@ -187,6 +96,7 @@
     "167": "English foxhound",
     "168": "redbone",
     "169": "borzoi, Russian wolfhound",
+    "17": "jay",
     "170": "Irish wolfhound",
     "171": "Italian greyhound",
     "172": "whippet",
@@ -197,6 +107,7 @@
     "177": "Scottish deerhound, deerhound",
     "178": "Weimaraner",
     "179": "Staffordshire bullterrier, Staffordshire bull terrier",
+    "18": "magpie",
     "180": "American Staffordshire terrier, Staffordshire terrier, American pit bull terrier, pit bull terrier",
     "181": "Bedlington terrier",
     "182": "Border terrier",
@@ -207,6 +118,7 @@
     "187": "Yorkshire terrier",
     "188": "wire-haired fox terrier",
     "189": "Lakeland terrier",
+    "19": "chickadee",
     "190": "Sealyham terrier, Sealyham",
     "191": "Airedale, Airedale terrier",
     "192": "cairn, cairn terrier",
@@ -217,6 +129,8 @@
     "197": "giant schnauzer",
     "198": "standard schnauzer",
     "199": "Scotch terrier, Scottish terrier, Scottie",
+    "2": "great white shark, white shark, man-eater, man-eating shark, Carcharodon carcharias",
+    "20": "water ouzel, dipper",
     "200": "Tibetan terrier, chrysanthemum dog",
     "201": "silky terrier, Sydney silky",
     "202": "soft-coated wheaten terrier",
@@ -227,6 +141,7 @@
     "207": "golden retriever",
     "208": "Labrador retriever",
     "209": "Chesapeake Bay retriever",
+    "21": "kite",
     "210": "German short-haired pointer",
     "211": "vizsla, Hungarian pointer",
     "212": "English setter",
@@ -237,6 +152,7 @@
     "217": "English springer, English springer spaniel",
     "218": "Welsh springer spaniel",
     "219": "cocker spaniel, English cocker spaniel, cocker",
+    "22": "bald eagle, American eagle, Haliaeetus leucocephalus",
     "220": "Sussex spaniel",
     "221": "Irish water spaniel",
     "222": "kuvasz",
@@ -247,6 +163,7 @@
     "227": "kelpie",
     "228": "komondor",
     "229": "Old English sheepdog, bobtail",
+    "23": "vulture",
     "230": "Shetland sheepdog, Shetland sheep dog, Shetland",
     "231": "collie",
     "232": "Border collie",
@@ -257,6 +174,7 @@
     "237": "miniature pinscher",
     "238": "Greater Swiss Mountain dog",
     "239": "Bernese mountain dog",
+    "24": "great grey owl, great gray owl, Strix nebulosa",
     "240": "Appenzeller",
     "241": "EntleBucher",
     "242": "boxer",
@@ -267,6 +185,7 @@
     "247": "Saint Bernard, St Bernard",
     "248": "Eskimo dog, husky",
     "249": "malamute, malemute, Alaskan malamute",
+    "25": "European fire salamander, Salamandra salamandra",
     "250": "Siberian husky",
     "251": "dalmatian, coach dog, carriage dog",
     "252": "affenpinscher, monkey pinscher, monkey dog",
@@ -277,6 +196,7 @@
     "257": "Great Pyrenees",
     "258": "Samoyed, Samoyede",
     "259": "Pomeranian",
+    "26": "common newt, Triturus vulgaris",
     "260": "chow, chow chow",
     "261": "keeshond",
     "262": "Brabancon griffon",
@@ -287,6 +207,7 @@
     "267": "standard poodle",
     "268": "Mexican hairless",
     "269": "timber wolf, grey wolf, gray wolf, Canis lupus",
+    "27": "eft",
     "270": "white wolf, Arctic wolf, Canis lupus tundrarum",
     "271": "red wolf, maned wolf, Canis rufus, Canis niger",
     "272": "coyote, prairie wolf, brush wolf, Canis latrans",
@@ -297,6 +218,7 @@
     "277": "red fox, Vulpes vulpes",
     "278": "kit fox, Vulpes macrotis",
     "279": "Arctic fox, white fox, Alopex lagopus",
+    "28": "spotted salamander, Ambystoma maculatum",
     "280": "grey fox, gray fox, Urocyon cinereoargenteus",
     "281": "tabby, tabby cat",
     "282": "tiger cat",
@@ -307,6 +229,7 @@
     "287": "lynx, catamount",
     "288": "leopard, Panthera pardus",
     "289": "snow leopard, ounce, Panthera uncia",
+    "29": "axolotl, mud puppy, Ambystoma mexicanum",
     "290": "jaguar, panther, Panthera onca, Felis onca",
     "291": "lion, king of beasts, Panthera leo",
     "292": "tiger, Panthera tigris",
@@ -317,6 +240,8 @@
     "297": "sloth bear, Melursus ursinus, Ursus ursinus",
     "298": "mongoose",
     "299": "meerkat, mierkat",
+    "3": "tiger shark, Galeocerdo cuvieri",
+    "30": "bullfrog, Rana catesbeiana",
     "300": "tiger beetle",
     "301": "ladybug, ladybeetle, lady beetle, ladybird, ladybird beetle",
     "302": "ground beetle, carabid beetle",
@@ -327,6 +252,7 @@
     "307": "weevil",
     "308": "fly",
     "309": "bee",
+    "31": "tree frog, tree-frog",
     "310": "ant, emmet, pismire",
     "311": "grasshopper, hopper",
     "312": "cricket",
@@ -337,6 +263,7 @@
     "317": "leafhopper",
     "318": "lacewing, lacewing fly",
     "319": "dragonfly, darning needle, devil's darning needle, sewing needle, snake feeder, snake doctor, mosquito hawk, skeeter hawk",
+    "32": "tailed frog, bell toad, ribbed toad, tailed toad, Ascaphus trui",
     "320": "damselfly",
     "321": "admiral",
     "322": "ringlet, ringlet butterfly",
@@ -347,6 +274,7 @@
     "327": "starfish, sea star",
     "328": "sea urchin",
     "329": "sea cucumber, holothurian",
+    "33": "loggerhead, loggerhead turtle, Caretta caretta",
     "330": "wood rabbit, cottontail, cottontail rabbit",
     "331": "hare",
     "332": "Angora, Angora rabbit",
@@ -357,6 +285,7 @@
     "337": "beaver",
     "338": "guinea pig, Cavia cobaya",
     "339": "sorrel",
+    "34": "leatherback turtle, leatherback, leathery turtle, Dermochelys coriacea",
     "340": "zebra",
     "341": "hog, pig, grunter, squealer, Sus scrofa",
     "342": "wild boar, boar, Sus scrofa",
@@ -367,6 +296,7 @@
     "347": "bison",
     "348": "ram, tup",
     "349": "bighorn, bighorn sheep, cimarron, Rocky Mountain bighorn, Rocky Mountain sheep, Ovis canadensis",
+    "35": "mud turtle",
     "350": "ibex, Capra ibex",
     "351": "hartebeest",
     "352": "impala, Aepyceros melampus",
@@ -377,6 +307,7 @@
     "357": "mink",
     "358": "polecat, fitch, foulmart, foumart, Mustela putorius",
     "359": "black-footed ferret, ferret, Mustela nigripes",
+    "36": "terrapin",
     "360": "otter",
     "361": "skunk, polecat, wood pussy",
     "362": "badger",
@@ -387,6 +318,7 @@
     "367": "chimpanzee, chimp, Pan troglodytes",
     "368": "gibbon, Hylobates lar",
     "369": "siamang, Hylobates syndactylus, Symphalangus syndactylus",
+    "37": "box turtle, box tortoise",
     "370": "guenon, guenon monkey",
     "371": "patas, hussar monkey, Erythrocebus patas",
     "372": "baboon",
@@ -397,6 +329,7 @@
     "377": "marmoset",
     "378": "capuchin, ringtail, Cebus capucinus",
     "379": "howler monkey, howler",
+    "38": "banded gecko",
     "380": "titi, titi monkey",
     "381": "spider monkey, Ateles geoffroyi",
     "382": "squirrel monkey, Saimiri sciureus",
@@ -407,6 +340,7 @@
     "387": "lesser panda, red panda, panda, bear cat, cat bear, Ailurus fulgens",
     "388": "giant panda, panda, panda bear, coon bear, Ailuropoda melanoleuca",
     "389": "barracouta, snoek",
+    "39": "common iguana, iguana, Iguana iguana",
     "390": "eel",
     "391": "coho, cohoe, coho salmon, blue jack, silver salmon, Oncorhynchus kisutch",
     "392": "rock beauty, Holocanthus tricolor",
@@ -417,6 +351,8 @@
     "397": "puffer, pufferfish, blowfish, globefish",
     "398": "abacus",
     "399": "abaya",
+    "4": "hammerhead, hammerhead shark",
+    "40": "American chameleon, anole, Anolis carolinensis",
     "400": "academic gown, academic robe, judge's robe",
     "401": "accordion, piano accordion, squeeze box",
     "402": "acoustic guitar",
@@ -427,6 +363,7 @@
     "407": "ambulance",
     "408": "amphibian, amphibious vehicle",
     "409": "analog clock",
+    "41": "whiptail, whiptail lizard",
     "410": "apiary, bee house",
     "411": "apron",
     "412": "ashcan, trash can, garbage can, wastebin, ash bin, ash-bin, ashbin, dustbin, trash barrel, trash bin",
@@ -437,6 +374,7 @@
     "417": "balloon",
     "418": "ballpoint, ballpoint pen, ballpen, Biro",
     "419": "Band Aid",
+    "42": "agama",
     "420": "banjo",
     "421": "bannister, banister, balustrade, balusters, handrail",
     "422": "barbell",
@@ -447,6 +385,7 @@
     "427": "barrel, cask",
     "428": "barrow, garden cart, lawn cart, wheelbarrow",
     "429": "baseball",
+    "43": "frilled lizard, Chlamydosaurus kingi",
     "430": "basketball",
     "431": "bassinet",
     "432": "bassoon",
@@ -457,6 +396,7 @@
     "437": "beacon, lighthouse, beacon light, pharos",
     "438": "beaker",
     "439": "bearskin, busby, shako",
+    "44": "alligator lizard",
     "440": "beer bottle",
     "441": "beer glass",
     "442": "bell cote, bell cot",
@@ -467,6 +407,7 @@
     "447": "binoculars, field glasses, opera glasses",
     "448": "birdhouse",
     "449": "boathouse",
+    "45": "Gila monster, Heloderma suspectum",
     "450": "bobsled, bobsleigh, bob",
     "451": "bolo tie, bolo, bola tie, bola",
     "452": "bonnet, poke bonnet",
@@ -477,6 +418,7 @@
     "457": "bow tie, bow-tie, bowtie",
     "458": "brass, memorial tablet, plaque",
     "459": "brassiere, bra, bandeau",
+    "46": "green lizard, Lacerta viridis",
     "460": "breakwater, groin, groyne, mole, bulwark, seawall, jetty",
     "461": "breastplate, aegis, egis",
     "462": "broom",
@@ -487,6 +429,7 @@
     "467": "butcher shop, meat market",
     "468": "cab, hack, taxi, taxicab",
     "469": "caldron, cauldron",
+    "47": "African chameleon, Chamaeleo chamaeleon",
     "470": "candle, taper, wax light",
     "471": "cannon",
     "472": "canoe",
@@ -497,6 +440,7 @@
     "477": "carpenter's kit, tool kit",
     "478": "carton",
     "479": "car wheel",
+    "48": "Komodo dragon, Komodo lizard, dragon lizard, giant lizard, Varanus komodoensis",
     "480": "cash machine, cash dispenser, automated teller machine, automatic teller machine, automated teller, automatic teller, ATM",
     "481": "cassette",
     "482": "cassette player",
@@ -507,6 +451,7 @@
     "487": "cellular telephone, cellular phone, cellphone, cell, mobile phone",
     "488": "chain",
     "489": "chainlink fence",
+    "49": "African crocodile, Nile crocodile, Crocodylus niloticus",
     "490": "chain mail, ring mail, mail, chain armor, chain armour, ring armor, ring armour",
     "491": "chain saw, chainsaw",
     "492": "chest",
@@ -517,6 +462,8 @@
     "497": "church, church building",
     "498": "cinema, movie theater, movie theatre, movie house, picture palace",
     "499": "cleaver, meat cleaver, chopper",
+    "5": "electric ray, crampfish, numbfish, torpedo",
+    "50": "American alligator, Alligator mississipiensis",
     "500": "cliff dwelling",
     "501": "cloak",
     "502": "clog, geta, patten, sabot",
@@ -527,6 +474,7 @@
     "507": "combination lock",
     "508": "computer keyboard, keypad",
     "509": "confectionery, confectionary, candy store",
+    "51": "triceratops",
     "510": "container ship, containership, container vessel",
     "511": "convertible",
     "512": "corkscrew, bottle screw",
@@ -537,6 +485,7 @@
     "517": "crane",
     "518": "crash helmet",
     "519": "crate",
+    "52": "thunder snake, worm snake, Carphophis amoenus",
     "520": "crib, cot",
     "521": "Crock Pot",
     "522": "croquet ball",
@@ -547,6 +496,7 @@
     "527": "desktop computer",
     "528": "dial telephone, dial phone",
     "529": "diaper, nappy, napkin",
+    "53": "ringneck snake, ring-necked snake, ring snake",
     "530": "digital clock",
     "531": "digital watch",
     "532": "dining table, board",
@@ -557,6 +507,7 @@
     "537": "dogsled, dog sled, dog sleigh",
     "538": "dome",
     "539": "doormat, welcome mat",
+    "54": "hognose snake, puff adder, sand viper",
     "540": "drilling platform, offshore rig",
     "541": "drum, membranophone, tympan",
     "542": "drumstick",
@@ -567,6 +518,7 @@
     "547": "electric locomotive",
     "548": "entertainment center",
     "549": "envelope",
+    "55": "green snake, grass snake",
     "550": "espresso maker",
     "551": "face powder",
     "552": "feather boa, boa",
@@ -577,6 +529,7 @@
     "557": "flagpole, flagstaff",
     "558": "flute, transverse flute",
     "559": "folding chair",
+    "56": "king snake, kingsnake",
     "560": "football helmet",
     "561": "forklift",
     "562": "fountain",
@@ -587,6 +540,7 @@
     "567": "frying pan, frypan, skillet",
     "568": "fur coat",
     "569": "garbage truck, dustcart",
+    "57": "garter snake, grass snake",
     "570": "gasmask, respirator, gas helmet",
     "571": "gas pump, gasoline pump, petrol pump, island dispenser",
     "572": "goblet",
@@ -597,6 +551,7 @@
     "577": "gong, tam-tam",
     "578": "gown",
     "579": "grand piano, grand",
+    "58": "water snake",
     "580": "greenhouse, nursery, glasshouse",
     "581": "grille, radiator grille",
     "582": "grocery store, grocery, food market, market",
@@ -607,6 +562,7 @@
     "587": "hammer",
     "588": "hamper",
     "589": "hand blower, blow dryer, blow drier, hair dryer, hair drier",
+    "59": "vine snake",
     "590": "hand-held computer, hand-held microcomputer",
     "591": "handkerchief, hankie, hanky, hankey",
     "592": "hard disc, hard disk, fixed disk",
@@ -617,6 +573,8 @@
     "597": "holster",
     "598": "home theater, home theatre",
     "599": "honeycomb",
+    "6": "stingray",
+    "60": "night snake, Hypsiglena torquata",
     "600": "hook, claw",
     "601": "hoopskirt, crinoline",
     "602": "horizontal bar, high bar",
@@ -627,6 +585,7 @@
     "607": "jack-o'-lantern",
     "608": "jean, blue jean, denim",
     "609": "jeep, landrover",
+    "61": "boa constrictor, Constrictor constrictor",
     "610": "jersey, T-shirt, tee shirt",
     "611": "jigsaw puzzle",
     "612": "jinrikisha, ricksha, rickshaw",
@@ -637,6 +596,7 @@
     "617": "lab coat, laboratory coat",
     "618": "ladle",
     "619": "lampshade, lamp shade",
+    "62": "rock python, rock snake, Python sebae",
     "620": "laptop, laptop computer",
     "621": "lawn mower, mower",
     "622": "lens cap, lens cover",
@@ -647,6 +607,7 @@
     "627": "limousine, limo",
     "628": "liner, ocean liner",
     "629": "lipstick, lip rouge",
+    "63": "Indian cobra, Naja naja",
     "630": "Loafer",
     "631": "lotion",
     "632": "loudspeaker, speaker, speaker unit, loudspeaker system, speaker system",
@@ -657,6 +618,7 @@
     "637": "mailbox, letter box",
     "638": "maillot",
     "639": "maillot, tank suit",
+    "64": "green mamba",
     "640": "manhole cover",
     "641": "maraca",
     "642": "marimba, xylophone",
@@ -667,6 +629,7 @@
     "647": "measuring cup",
     "648": "medicine chest, medicine cabinet",
     "649": "megalith, megalithic structure",
+    "65": "sea snake",
     "650": "microphone, mike",
     "651": "microwave, microwave oven",
     "652": "military uniform",
@@ -677,6 +640,7 @@
     "657": "missile",
     "658": "mitten",
     "659": "mixing bowl",
+    "66": "horned viper, cerastes, sand viper, horned asp, Cerastes cornutus",
     "660": "mobile home, manufactured home",
     "661": "Model T",
     "662": "modem",
@@ -687,6 +651,7 @@
     "667": "mortarboard",
     "668": "mosque",
     "669": "mosquito net",
+    "67": "diamondback, diamondback rattlesnake, Crotalus adamanteus",
     "670": "motor scooter, scooter",
     "671": "mountain bike, all-terrain bike, off-roader",
     "672": "mountain tent",
@@ -697,6 +662,7 @@
     "677": "nail",
     "678": "neck brace",
     "679": "necklace",
+    "68": "sidewinder, horned rattlesnake, Crotalus cerastes",
     "680": "nipple",
     "681": "notebook, notebook computer",
     "682": "obelisk",
@@ -707,6 +673,7 @@
     "687": "organ, pipe organ",
     "688": "oscilloscope, scope, cathode-ray oscilloscope, CRO",
     "689": "overskirt",
+    "69": "trilobite",
     "690": "oxcart",
     "691": "oxygen mask",
     "692": "packet",
@@ -717,6 +684,8 @@
     "697": "pajama, pyjama, pj's, jammies",
     "698": "palace",
     "699": "panpipe, pandean pipe, syrinx",
+    "7": "cock",
+    "70": "harvestman, daddy longlegs, Phalangium opilio",
     "700": "paper towel",
     "701": "parachute, chute",
     "702": "parallel bars, bars",
@@ -727,6 +696,7 @@
     "707": "pay-phone, pay-station",
     "708": "pedestal, plinth, footstall",
     "709": "pencil box, pencil case",
+    "71": "scorpion",
     "710": "pencil sharpener",
     "711": "perfume, essence",
     "712": "Petri dish",
@@ -737,6 +707,7 @@
     "717": "pickup, pickup truck",
     "718": "pier",
     "719": "piggy bank, penny bank",
+    "72": "black and gold garden spider, Argiope aurantia",
     "720": "pill bottle",
     "721": "pillow",
     "722": "ping-pong ball",
@@ -747,6 +718,7 @@
     "727": "planetarium",
     "728": "plastic bag",
     "729": "plate rack",
+    "73": "barn spider, Araneus cavaticus",
     "730": "plow, plough",
     "731": "plunger, plumber's helper",
     "732": "Polaroid camera, Polaroid Land camera",
@@ -757,6 +729,7 @@
     "737": "pop bottle, soda bottle",
     "738": "pot, flowerpot",
     "739": "potter's wheel",
+    "74": "garden spider, Aranea diademata",
     "740": "power drill",
     "741": "prayer rug, prayer mat",
     "742": "printer",
@@ -767,6 +740,7 @@
     "747": "punching bag, punch bag, punching ball, punchball",
     "748": "purse",
     "749": "quill, quill pen",
+    "75": "black widow, Latrodectus mactans",
     "750": "quilt, comforter, comfort, puff",
     "751": "racer, race car, racing car",
     "752": "racket, racquet",
@@ -777,6 +751,7 @@
     "757": "recreational vehicle, RV, R.V.",
     "758": "reel",
     "759": "reflex camera",
+    "76": "tarantula",
     "760": "refrigerator, icebox",
     "761": "remote control, remote",
     "762": "restaurant, eating house, eating place, eatery",
@@ -787,6 +762,7 @@
     "767": "rubber eraser, rubber, pencil eraser",
     "768": "rugby ball",
     "769": "rule, ruler",
+    "77": "wolf spider, hunting spider",
     "770": "running shoe",
     "771": "safe",
     "772": "safety pin",
@@ -797,6 +773,7 @@
     "777": "scabbard",
     "778": "scale, weighing machine",
     "779": "school bus",
+    "78": "tick",
     "780": "schooner",
     "781": "scoreboard",
     "782": "screen, CRT screen",
@@ -807,6 +784,7 @@
     "787": "shield, buckler",
     "788": "shoe shop, shoe-shop, shoe store",
     "789": "shoji",
+    "79": "centipede",
     "790": "shopping basket",
     "791": "shopping cart",
     "792": "shovel",
@@ -817,6 +795,8 @@
     "797": "sleeping bag",
     "798": "slide rule, slipstick",
     "799": "sliding door",
+    "8": "hen",
+    "80": "black grouse",
     "800": "slot, one-armed bandit",
     "801": "snorkel",
     "802": "snowmobile",
@@ -827,6 +807,7 @@
     "807": "solar dish, solar collector, solar furnace",
     "808": "sombrero",
     "809": "soup bowl",
+    "81": "ptarmigan",
     "810": "space bar",
     "811": "space heater",
     "812": "space shuttle",
@@ -837,6 +818,7 @@
     "817": "sports car, sport car",
     "818": "spotlight, spot",
     "819": "stage",
+    "82": "ruffed grouse, partridge, Bonasa umbellus",
     "820": "steam locomotive",
     "821": "steel arch bridge",
     "822": "steel drum",
@@ -847,6 +829,7 @@
     "827": "stove",
     "828": "strainer",
     "829": "streetcar, tram, tramcar, trolley, trolley car",
+    "83": "prairie chicken, prairie grouse, prairie fowl",
     "830": "stretcher",
     "831": "studio couch, day bed",
     "832": "stupa, tope",
@@ -857,6 +840,7 @@
     "837": "sunglasses, dark glasses, shades",
     "838": "sunscreen, sunblock, sun blocker",
     "839": "suspension bridge",
+    "84": "peacock",
     "840": "swab, swob, mop",
     "841": "sweatshirt",
     "842": "swimming trunks, bathing trunks",
@@ -867,6 +851,7 @@
     "847": "tank, army tank, armored combat vehicle, armoured combat vehicle",
     "848": "tape player",
     "849": "teapot",
+    "85": "quail",
     "850": "teddy, teddy bear",
     "851": "television, television system",
     "852": "tennis ball",
@@ -877,6 +862,7 @@
     "857": "throne",
     "858": "tile roof",
     "859": "toaster",
+    "86": "partridge",
     "860": "tobacco shop, tobacconist shop, tobacconist",
     "861": "toilet seat",
     "862": "torch",
@@ -887,6 +873,7 @@
     "867": "trailer truck, tractor trailer, trucking rig, rig, articulated lorry, semi",
     "868": "tray",
     "869": "trench coat",
+    "87": "African grey, African gray, Psittacus erithacus",
     "870": "tricycle, trike, velocipede",
     "871": "trimaran",
     "872": "tripod",
@@ -897,6 +884,7 @@
     "877": "turnstile",
     "878": "typewriter keyboard",
     "879": "umbrella",
+    "88": "macaw",
     "880": "unicycle, monocycle",
     "881": "upright, upright piano",
     "882": "vacuum, vacuum cleaner",
@@ -907,6 +895,7 @@
     "887": "vestment",
     "888": "viaduct",
     "889": "violin, fiddle",
+    "89": "sulphur-crested cockatoo, Kakatoe galerita, Cacatua galerita",
     "890": "volleyball",
     "891": "waffle iron",
     "892": "wall clock",
@@ -917,6 +906,8 @@
     "897": "washer, automatic washer, washing machine",
     "898": "water bottle",
     "899": "water jug",
+    "9": "ostrich, Struthio camelus",
+    "90": "lorikeet",
     "900": "water tower",
     "901": "whiskey jug",
     "902": "whistle",
@@ -927,6 +918,7 @@
     "907": "wine bottle",
     "908": "wing",
     "909": "wok",
+    "91": "coucal",
     "910": "wooden spoon",
     "911": "wool, woolen, woollen",
     "912": "worm fence, snake fence, snake-rail fence, Virginia fence",
@@ -937,6 +929,7 @@
     "917": "comic book",
     "918": "crossword puzzle, crossword",
     "919": "street sign",
+    "92": "bee eater",
     "920": "traffic light, traffic signal, stoplight",
     "921": "book jacket, dust cover, dust jacket, dust wrapper",
     "922": "menu",
@@ -947,6 +940,7 @@
     "927": "trifle",
     "928": "ice cream, icecream",
     "929": "ice lolly, lolly, lollipop, popsicle",
+    "93": "hornbill",
     "930": "French loaf",
     "931": "bagel, beigel",
     "932": "pretzel",
@@ -957,6 +951,7 @@
     "937": "broccoli",
     "938": "cauliflower",
     "939": "zucchini, courgette",
+    "94": "hummingbird",
     "940": "spaghetti squash",
     "941": "acorn squash",
     "942": "butternut squash",
@@ -967,6 +962,7 @@
     "947": "mushroom",
     "948": "Granny Smith",
     "949": "strawberry",
+    "95": "jacamar",
     "950": "orange",
     "951": "lemon",
     "952": "fig",
@@ -977,6 +973,7 @@
     "957": "pomegranate",
     "958": "hay",
     "959": "carbonara",
+    "96": "toucan",
     "960": "chocolate sauce, chocolate syrup",
     "961": "dough",
     "962": "meat loaf, meatloaf",
@@ -987,6 +984,7 @@
     "967": "espresso",
     "968": "cup",
     "969": "eggnog",
+    "97": "drake",
     "970": "alp",
     "971": "bubble",
     "972": "cliff, drop, drop-off",
@@ -997,6 +995,7 @@
     "977": "sandbar, sand bar",
     "978": "seashore, coast, seacoast, sea-coast",
     "979": "valley, vale",
+    "98": "red-breasted merganser, Mergus serrator",
     "980": "volcano",
     "981": "ballplayer, baseball player",
     "982": "groom, bridegroom",
@@ -1007,6 +1006,7 @@
     "987": "corn",
     "988": "acorn",
     "989": "hip, rose hip, rosehip",
+    "99": "goose",
     "990": "buckeye, horse chestnut, conker",
     "991": "coral fungus",
     "992": "agaric",
@@ -2052,4 +2052,4 @@
   "transformers_version": "4.44.0",
   "use_absolute_embeddings": false,
   "window_size": 7
-}
+}