|
{ |
|
"_name_or_path": "custom/dog-breed-classifier", |
|
"architectures": [ |
|
"ResNetForImageClassification" |
|
], |
|
"attention_probs_dropout_prob": 0.0, |
|
"hidden_act": "relu", |
|
"hidden_dropout_prob": 0.0, |
|
"hidden_size": 2048, |
|
"id2label": { |
|
"0": " Afghan Hound", |
|
"1": " Akita", |
|
"2": " Alaskan Malamute", |
|
"3": " Australian Cattle Dog", |
|
"4": " Australian Shepherd", |
|
"5": " Basenji", |
|
"6": " Basset Hound", |
|
"7": " Beagle", |
|
"8": " Belgian Malinois", |
|
"9": " Bernese Mountain Dog", |
|
"10": " Bichon Frise", |
|
"11": " Bloodhound", |
|
"12": " Border Collie", |
|
"13": " Boston Terrier", |
|
"14": " Boxer", |
|
"15": " Brittany", |
|
"16": " Bull Terrier", |
|
"17": " Bulldog", |
|
"18": " Bullmastiff", |
|
"19": " Cairn Terrier", |
|
"20": " Cavalier King Charles Spaniel", |
|
"21": " Chesapeake Bay Retriever", |
|
"22": " Chihuahua", |
|
"23": " Chinese Shar-Pei", |
|
"24": " Collie", |
|
"25": " Dachshund", |
|
"26": " Dalmatian", |
|
"27": " Doberman Pinscher", |
|
"28": " English Mastiff", |
|
"29": " English Springer Spaniel", |
|
"30": " German Shepherd", |
|
"31": " Golden Retriever", |
|
"32": " Great Dane", |
|
"33": " Havanese", |
|
"34": " Irish Setter", |
|
"35": " Italian Greyhound", |
|
"36": " Labrador Retriever", |
|
"37": " Lhasa Apso", |
|
"38": " Maltese", |
|
"39": " Miniature Schnauzer", |
|
"40": " Newfoundland", |
|
"41": " Papillon", |
|
"42": " Pembroke Welsh Corgi", |
|
"43": " Poodle", |
|
"44": " Portuguese Water Dog", |
|
"45": " Rhodesian Ridgeback", |
|
"46": " Rottweiler", |
|
"47": " Saint Bernard", |
|
"48": " Samoyed", |
|
"49": " Shetland Sheepdog", |
|
"50": " Shiba Inu", |
|
"51": " Shih Tzu", |
|
"52": " Siberian Husky", |
|
"53": " Staffordshire Bull Terrier", |
|
"54": " Vizsla", |
|
"55": " Weimaraner", |
|
"56": " West Highland White Terrier", |
|
"57": " Whippet", |
|
"58": " Wire Fox Terrier", |
|
"59": " Yorkshire Terrier" |
|
}, |
|
"image_size": 224, |
|
"initializer_range": 0.02, |
|
"intermediate_size": 4096, |
|
"label2id": { |
|
" Afghan Hound": "0", |
|
" Akita": "1", |
|
" Alaskan Malamute": "2", |
|
" Australian Cattle Dog": "3", |
|
" Australian Shepherd": "4", |
|
" Basenji": "5", |
|
" Basset Hound": "6", |
|
" Beagle": "7", |
|
" Belgian Malinois": "8", |
|
" Bernese Mountain Dog": "9", |
|
" Bichon Frise": "10", |
|
" Bloodhound": "11", |
|
" Border Collie": "12", |
|
" Boston Terrier": "13", |
|
" Boxer": "14", |
|
" Brittany": "15", |
|
" Bull Terrier": "16", |
|
" Bulldog": "17", |
|
" Bullmastiff": "18", |
|
" Cairn Terrier": "19", |
|
" Cavalier King Charles Spaniel": "20", |
|
" Chesapeake Bay Retriever": "21", |
|
" Chihuahua": "22", |
|
" Chinese Shar-Pei": "23", |
|
" Collie": "24", |
|
" Dachshund": "25", |
|
" Dalmatian": "26", |
|
" Doberman Pinscher": "27", |
|
" English Mastiff": "28", |
|
" English Springer Spaniel": "29", |
|
" German Shepherd": "30", |
|
" Golden Retriever": "31", |
|
" Great Dane": "32", |
|
" Havanese": "33", |
|
" Irish Setter": "34", |
|
" Italian Greyhound": "35", |
|
" Labrador Retriever": "36", |
|
" Lhasa Apso": "37", |
|
" Maltese": "38", |
|
" Miniature Schnauzer": "39", |
|
" Newfoundland": "40", |
|
" Papillon": "41", |
|
" Pembroke Welsh Corgi": "42", |
|
" Poodle": "43", |
|
" Portuguese Water Dog": "44", |
|
" Rhodesian Ridgeback": "45", |
|
" Rottweiler": "46", |
|
" Saint Bernard": "47", |
|
" Samoyed": "48", |
|
" Shetland Sheepdog": "49", |
|
" Shiba Inu": "50", |
|
" Shih Tzu": "51", |
|
" Siberian Husky": "52", |
|
" Staffordshire Bull Terrier": "53", |
|
" Vizsla": "54", |
|
" Weimaraner": "55", |
|
" West Highland White Terrier": "56", |
|
" Whippet": "57", |
|
" Wire Fox Terrier": "58", |
|
" Yorkshire Terrier": "59" |
|
}, |
|
"layer_norm_eps": 1e-12, |
|
"model_type": "resnet", |
|
"num_attention_heads": 32, |
|
"num_channels": 3, |
|
"num_hidden_layers": 50, |
|
"patch_size": 16, |
|
"problem_type": "single_label_classification", |
|
"qkv_bias": true, |
|
"torch_dtype": "float32", |
|
"transformers_version": "4.38.2" |
|
} |