dog-breed-classifier / config.json
valentinocc's picture
Update config.json
9bb4e93 verified
raw
history blame
6.87 kB
{
"_name_or_path": "valentinocc/dog-breed-classifier",
"architectures": [
"MobileNetV2ForImageClassification"
],
"classifier_dropout": 0.2,
"hidden_act": "relu6",
"hidden_size": 1280,
"image_size": 224,
"initializer_range": 0.02,
"layer_norm_eps": 1e-05,
"model_type": "mobilenet_v2",
"num_channels": 3,
"num_labels": 120,
"output_stride": 32,
"problem_type": "single_label_classification",
"torch_dtype": "float32",
"transformers_version": "4.21.0",
"width_mult": 1.0,
"id2label": {
"0": "Chihuahua",
"1": "Japanese Spaniel",
"2": "Maltese Dog",
"3": "Pekinese",
"4": "Shih-Tzu",
"5": "Blenheim Spaniel",
"6": "Papillon",
"7": "Toy Terrier",
"8": "Rhodesian Ridgeback",
"9": "Afghan Hound",
"10": "Basset",
"11": "Beagle",
"12": "Bloodhound",
"13": "Bluetick",
"14": "Black-And-Tan Coonhound",
"15": "Walker Hound",
"16": "English Foxhound",
"17": "Redbone",
"18": "Borzoi",
"19": "Irish Wolfhound",
"20": "Italian Greyhound",
"21": "Whippet",
"22": "Ibizan Hound",
"23": "Norwegian Elkhound",
"24": "Otterhound",
"25": "Saluki",
"26": "Scottish Deerhound",
"27": "Weimaraner",
"28": "Staffordshire Bullterrier",
"29": "American Staffordshire Terrier",
"30": "Bedlington Terrier",
"31": "Border Terrier",
"32": "Kerry Blue Terrier",
"33": "Irish Terrier",
"34": "Norfolk Terrier",
"35": "Norwich Terrier",
"36": "Yorkshire Terrier",
"37": "Wire-Haired Fox Terrier",
"38": "Lakeland Terrier",
"39": "Sealyham Terrier",
"40": "Airedale",
"41": "Cairn",
"42": "Australian Terrier",
"43": "Dandie Dinmont",
"44": "Boston Bull",
"45": "Miniature Schnauzer",
"46": "Giant Schnauzer",
"47": "Standard Schnauzer",
"48": "Scotch Terrier",
"49": "Tibetan Terrier",
"50": "Silky Terrier",
"51": "Soft-Coated Wheaten Terrier",
"52": "West Highland White Terrier",
"53": "Lhasa",
"54": "Flat-Coated Retriever",
"55": "Curly-Coated Retriever",
"56": "Golden Retriever",
"57": "Labrador Retriever",
"58": "Chesapeake Bay Retriever",
"59": "German Short-Haired Pointer",
"60": "Vizsla",
"61": "English Setter",
"62": "Irish Setter",
"63": "Gordon Setter",
"64": "Brittany Spaniel",
"65": "Clumber",
"66": "English Springer",
"67": "Welsh Springer Spaniel",
"68": "Cocker Spaniel",
"69": "Sussex Spaniel",
"70": "Irish Water Spaniel",
"71": "Kuvasz",
"72": "Schipperke",
"73": "Groenendael",
"74": "Malinois",
"75": "Briard",
"76": "Kelpie",
"77": "Komondor",
"78": "Old English Sheepdog",
"79": "Shetland Sheepdog",
"80": "Collie",
"81": "Border Collie",
"82": "Bouvier Des Flandres",
"83": "Rottweiler",
"84": "German Shepherd",
"85": "Doberman",
"86": "Miniature Pinscher",
"87": "Greater Swiss Mountain Dog",
"88": "Bernese Mountain Dog",
"89": "Appenzeller",
"90": "Entlebucher",
"91": "Boxer",
"92": "Bull Mastiff",
"93": "Tibetan Mastiff",
"94": "French Bulldog",
"95": "Great Dane",
"96": "Saint Bernard",
"97": "Eskimo Dog",
"98": "Malamute",
"99": "Siberian Husky",
"100": "Affenpinscher",
"101": "Basenji",
"102": "Pug",
"103": "Leonberg",
"104": "Newfoundland",
"105": "Great Pyrenees",
"106": "Samoyed",
"107": "Pomeranian",
"108": "Chow",
"109": "Keeshond",
"110": "Brabancon Griffon",
"111": "Pembroke",
"112": "Cardigan",
"113": "Toy Poodle",
"114": "Miniature Poodle",
"115": "Standard Poodle",
"116": "Mexican Hairless",
"117": "Dingo",
"118": "Dhole",
"119": "African Hunting Dog"
},
"label2id": {
"Chihuahua": 0,
"Japanese Spaniel": 1,
"Maltese Dog": 2,
"Pekinese": 3,
"Shih-Tzu": 4,
"Blenheim Spaniel": 5,
"Papillon": 6,
"Toy Terrier": 7,
"Rhodesian Ridgeback": 8,
"Afghan Hound": 9,
"Basset": 10,
"Beagle": 11,
"Bloodhound": 12,
"Bluetick": 13,
"Black-And-Tan Coonhound": 14,
"Walker Hound": 15,
"English Foxhound": 16,
"Redbone": 17,
"Borzoi": 18,
"Irish Wolfhound": 19,
"Italian Greyhound": 20,
"Whippet": 21,
"Ibizan Hound": 22,
"Norwegian Elkhound": 23,
"Otterhound": 24,
"Saluki": 25,
"Scottish Deerhound": 26,
"Weimaraner": 27,
"Staffordshire Bullterrier": 28,
"American Staffordshire Terrier": 29,
"Bedlington Terrier": 30,
"Border Terrier": 31,
"Kerry Blue Terrier": 32,
"Irish Terrier": 33,
"Norfolk Terrier": 34,
"Norwich Terrier": 35,
"Yorkshire Terrier": 36,
"Wire-Haired Fox Terrier": 37,
"Lakeland Terrier": 38,
"Sealyham Terrier": 39,
"Airedale": 40,
"Cairn": 41,
"Australian Terrier": 42,
"Dandie Dinmont": 43,
"Boston Bull": 44,
"Miniature Schnauzer": 45,
"Giant Schnauzer": 46,
"Standard Schnauzer": 47,
"Scotch Terrier": 48,
"Tibetan Terrier": 49,
"Silky Terrier": 50,
"Soft-Coated Wheaten Terrier": 51,
"West Highland White Terrier": 52,
"Lhasa": 53,
"Flat-Coated Retriever": 54,
"Curly-Coated Retriever": 55,
"Golden Retriever": 56,
"Labrador Retriever": 57,
"Chesapeake Bay Retriever": 58,
"German Short-Haired Pointer": 59,
"Vizsla": 60,
"English Setter": 61,
"Irish Setter": 62,
"Gordon Setter": 63,
"Brittany Spaniel": 64,
"Clumber": 65,
"English Springer": 66,
"Welsh Springer Spaniel": 67,
"Cocker Spaniel": 68,
"Sussex Spaniel": 69,
"Irish Water Spaniel": 70,
"Kuvasz": 71,
"Schipperke": 72,
"Groenendael": 73,
"Malinois": 74,
"Briard": 75,
"Kelpie": 76,
"Komondor": 77,
"Old English Sheepdog": 78,
"Shetland Sheepdog": 79,
"Collie": 80,
"Border Collie": 81,
"Bouvier Des Flandres": 82,
"Rottweiler": 83,
"German Shepherd": 84,
"Doberman": 85,
"Miniature Pinscher": 86,
"Greater Swiss Mountain Dog": 87,
"Bernese Mountain Dog": 88,
"Appenzeller": 89,
"Entlebucher": 90,
"Boxer": 91,
"Bull Mastiff": 92,
"Tibetan Mastiff": 93,
"French Bulldog": 94,
"Great Dane": 95,
"Saint Bernard": 96,
"Eskimo Dog": 97,
"Malamute": 98,
"Siberian Husky": 99,
"Affenpinscher": 100,
"Basenji": 101,
"Pug": 102,
"Leonberg": 103,
"Newfoundland": 104,
"Great Pyrenees": 105,
"Samoyed": 106,
"Pomeranian": 107,
"Chow": 108,
"Keeshond": 109,
"Brabancon Griffon": 110,
"Pembroke": 111,
"Cardigan": 112,
"Toy Poodle": 113,
"Miniature Poodle": 114,
"Standard Poodle": 115,
"Mexican Hairless": 116,
"Dingo": 117,
"Dhole": 118,
"African Hunting Dog": 119
}
}