End of training
Browse files- README.md +174 -0
- config.json +285 -0
- model.safetensors +3 -0
- preprocessor_config.json +22 -0
- training_args.bin +3 -0
README.md
ADDED
@@ -0,0 +1,174 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: apache-2.0
|
3 |
+
base_model: microsoft/resnet-101
|
4 |
+
tags:
|
5 |
+
- generated_from_trainer
|
6 |
+
datasets:
|
7 |
+
- imagefolder
|
8 |
+
metrics:
|
9 |
+
- accuracy
|
10 |
+
model-index:
|
11 |
+
- name: Dogs-Breed-Image-Classification-V1
|
12 |
+
results:
|
13 |
+
- task:
|
14 |
+
name: Image Classification
|
15 |
+
type: image-classification
|
16 |
+
dataset:
|
17 |
+
name: imagefolder
|
18 |
+
type: imagefolder
|
19 |
+
config: default
|
20 |
+
split: train
|
21 |
+
args: default
|
22 |
+
metrics:
|
23 |
+
- name: Accuracy
|
24 |
+
type: accuracy
|
25 |
+
value: 0.8757971454600668
|
26 |
+
---
|
27 |
+
|
28 |
+
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
29 |
+
should probably proofread and complete it, then remove this comment. -->
|
30 |
+
|
31 |
+
# Dogs-Breed-Image-Classification-V1
|
32 |
+
|
33 |
+
This model is a fine-tuned version of [microsoft/resnet-101](https://huggingface.co/microsoft/resnet-101) on the imagefolder dataset.
|
34 |
+
It achieves the following results on the evaluation set:
|
35 |
+
- Loss: 0.4469
|
36 |
+
- Accuracy: 0.8758
|
37 |
+
|
38 |
+
## Model description
|
39 |
+
|
40 |
+
More information needed
|
41 |
+
|
42 |
+
## Intended uses & limitations
|
43 |
+
|
44 |
+
More information needed
|
45 |
+
|
46 |
+
## Training and evaluation data
|
47 |
+
|
48 |
+
More information needed
|
49 |
+
|
50 |
+
## Training procedure
|
51 |
+
|
52 |
+
### Training hyperparameters
|
53 |
+
|
54 |
+
The following hyperparameters were used during training:
|
55 |
+
- learning_rate: 1e-05
|
56 |
+
- train_batch_size: 32
|
57 |
+
- eval_batch_size: 32
|
58 |
+
- seed: 42
|
59 |
+
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
60 |
+
- lr_scheduler_type: linear
|
61 |
+
- num_epochs: 100
|
62 |
+
|
63 |
+
### Training results
|
64 |
+
|
65 |
+
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|
66 |
+
|:-------------:|:-----:|:-----:|:---------------:|:--------:|
|
67 |
+
| No log | 1.0 | 309 | 18.7685 | 0.0091 |
|
68 |
+
| 18.7211 | 2.0 | 618 | 18.5975 | 0.0091 |
|
69 |
+
| 18.7211 | 3.0 | 927 | 17.4087 | 0.0091 |
|
70 |
+
| 15.4274 | 4.0 | 1236 | 11.8712 | 0.0091 |
|
71 |
+
| 10.3252 | 5.0 | 1545 | 6.6642 | 0.0091 |
|
72 |
+
| 10.3252 | 6.0 | 1854 | 5.2754 | 0.0112 |
|
73 |
+
| 6.2268 | 7.0 | 2163 | 4.8454 | 0.0158 |
|
74 |
+
| 6.2268 | 8.0 | 2472 | 4.7658 | 0.0140 |
|
75 |
+
| 4.9682 | 9.0 | 2781 | 4.6860 | 0.0234 |
|
76 |
+
| 4.7245 | 10.0 | 3090 | 4.6165 | 0.0316 |
|
77 |
+
| 4.7245 | 11.0 | 3399 | 4.5349 | 0.0446 |
|
78 |
+
| 4.5441 | 12.0 | 3708 | 4.4555 | 0.0623 |
|
79 |
+
| 4.3912 | 13.0 | 4017 | 4.3437 | 0.0862 |
|
80 |
+
| 4.3912 | 14.0 | 4326 | 4.2182 | 0.1330 |
|
81 |
+
| 4.2211 | 15.0 | 4635 | 4.0752 | 0.2153 |
|
82 |
+
| 4.2211 | 16.0 | 4944 | 3.9803 | 0.2599 |
|
83 |
+
| 3.9762 | 17.0 | 5253 | 3.7347 | 0.3596 |
|
84 |
+
| 3.69 | 18.0 | 5562 | 3.5493 | 0.4194 |
|
85 |
+
| 3.69 | 19.0 | 5871 | 3.3404 | 0.4813 |
|
86 |
+
| 3.3803 | 20.0 | 6180 | 3.1122 | 0.5600 |
|
87 |
+
| 3.3803 | 21.0 | 6489 | 2.8656 | 0.6101 |
|
88 |
+
| 3.0345 | 22.0 | 6798 | 2.6544 | 0.6462 |
|
89 |
+
| 2.6793 | 23.0 | 7107 | 2.4178 | 0.6647 |
|
90 |
+
| 2.6793 | 24.0 | 7416 | 2.1967 | 0.7121 |
|
91 |
+
| 2.3251 | 25.0 | 7725 | 2.0091 | 0.7203 |
|
92 |
+
| 1.9975 | 26.0 | 8034 | 1.8189 | 0.7464 |
|
93 |
+
| 1.9975 | 27.0 | 8343 | 1.6537 | 0.7519 |
|
94 |
+
| 1.7009 | 28.0 | 8652 | 1.4413 | 0.7880 |
|
95 |
+
| 1.7009 | 29.0 | 8961 | 1.3137 | 0.7968 |
|
96 |
+
| 1.4494 | 30.0 | 9270 | 1.2150 | 0.7929 |
|
97 |
+
| 1.2389 | 31.0 | 9579 | 1.1238 | 0.8041 |
|
98 |
+
| 1.2389 | 32.0 | 9888 | 1.0215 | 0.8208 |
|
99 |
+
| 1.0646 | 33.0 | 10197 | 0.9637 | 0.8190 |
|
100 |
+
| 0.9319 | 34.0 | 10506 | 0.8891 | 0.8299 |
|
101 |
+
| 0.9319 | 35.0 | 10815 | 0.8520 | 0.8330 |
|
102 |
+
| 0.8297 | 36.0 | 11124 | 0.8212 | 0.8400 |
|
103 |
+
| 0.8297 | 37.0 | 11433 | 0.7579 | 0.8415 |
|
104 |
+
| 0.7293 | 38.0 | 11742 | 0.7254 | 0.8454 |
|
105 |
+
| 0.6657 | 39.0 | 12051 | 0.7019 | 0.8457 |
|
106 |
+
| 0.6657 | 40.0 | 12360 | 0.6669 | 0.8527 |
|
107 |
+
| 0.6047 | 41.0 | 12669 | 0.6510 | 0.8530 |
|
108 |
+
| 0.6047 | 42.0 | 12978 | 0.6264 | 0.8545 |
|
109 |
+
| 0.557 | 43.0 | 13287 | 0.6275 | 0.8506 |
|
110 |
+
| 0.5126 | 44.0 | 13596 | 0.5947 | 0.8536 |
|
111 |
+
| 0.5126 | 45.0 | 13905 | 0.5860 | 0.8573 |
|
112 |
+
| 0.475 | 46.0 | 14214 | 0.5745 | 0.8545 |
|
113 |
+
| 0.4406 | 47.0 | 14523 | 0.5579 | 0.8600 |
|
114 |
+
| 0.4406 | 48.0 | 14832 | 0.5386 | 0.8621 |
|
115 |
+
| 0.4086 | 49.0 | 15141 | 0.5346 | 0.8624 |
|
116 |
+
| 0.4086 | 50.0 | 15450 | 0.5200 | 0.8612 |
|
117 |
+
| 0.3882 | 51.0 | 15759 | 0.5233 | 0.8612 |
|
118 |
+
| 0.3646 | 52.0 | 16068 | 0.5148 | 0.8640 |
|
119 |
+
| 0.3646 | 53.0 | 16377 | 0.5078 | 0.8679 |
|
120 |
+
| 0.3386 | 54.0 | 16686 | 0.5067 | 0.8646 |
|
121 |
+
| 0.3386 | 55.0 | 16995 | 0.4976 | 0.8673 |
|
122 |
+
| 0.3208 | 56.0 | 17304 | 0.4934 | 0.8682 |
|
123 |
+
| 0.3039 | 57.0 | 17613 | 0.4849 | 0.8688 |
|
124 |
+
| 0.3039 | 58.0 | 17922 | 0.4930 | 0.8691 |
|
125 |
+
| 0.2915 | 59.0 | 18231 | 0.4867 | 0.8655 |
|
126 |
+
| 0.2784 | 60.0 | 18540 | 0.4832 | 0.8679 |
|
127 |
+
| 0.2784 | 61.0 | 18849 | 0.4785 | 0.8670 |
|
128 |
+
| 0.2597 | 62.0 | 19158 | 0.4753 | 0.8685 |
|
129 |
+
| 0.2597 | 63.0 | 19467 | 0.4701 | 0.8712 |
|
130 |
+
| 0.2488 | 64.0 | 19776 | 0.4766 | 0.8697 |
|
131 |
+
| 0.2426 | 65.0 | 20085 | 0.4726 | 0.8700 |
|
132 |
+
| 0.2426 | 66.0 | 20394 | 0.4670 | 0.8694 |
|
133 |
+
| 0.2261 | 67.0 | 20703 | 0.4624 | 0.8722 |
|
134 |
+
| 0.2252 | 68.0 | 21012 | 0.4631 | 0.8718 |
|
135 |
+
| 0.2252 | 69.0 | 21321 | 0.4702 | 0.8670 |
|
136 |
+
| 0.2116 | 70.0 | 21630 | 0.4629 | 0.8715 |
|
137 |
+
| 0.2116 | 71.0 | 21939 | 0.4650 | 0.8685 |
|
138 |
+
| 0.2032 | 72.0 | 22248 | 0.4670 | 0.8673 |
|
139 |
+
| 0.2035 | 73.0 | 22557 | 0.4565 | 0.8670 |
|
140 |
+
| 0.2035 | 74.0 | 22866 | 0.4550 | 0.8697 |
|
141 |
+
| 0.19 | 75.0 | 23175 | 0.4544 | 0.8706 |
|
142 |
+
| 0.19 | 76.0 | 23484 | 0.4483 | 0.8670 |
|
143 |
+
| 0.1833 | 77.0 | 23793 | 0.4650 | 0.8694 |
|
144 |
+
| 0.184 | 78.0 | 24102 | 0.4604 | 0.8709 |
|
145 |
+
| 0.184 | 79.0 | 24411 | 0.4484 | 0.8697 |
|
146 |
+
| 0.1728 | 80.0 | 24720 | 0.4469 | 0.8758 |
|
147 |
+
| 0.1688 | 81.0 | 25029 | 0.4536 | 0.8676 |
|
148 |
+
| 0.1688 | 82.0 | 25338 | 0.4450 | 0.8709 |
|
149 |
+
| 0.1674 | 83.0 | 25647 | 0.4530 | 0.8691 |
|
150 |
+
| 0.1674 | 84.0 | 25956 | 0.4532 | 0.8725 |
|
151 |
+
| 0.1632 | 85.0 | 26265 | 0.4495 | 0.8718 |
|
152 |
+
| 0.1605 | 86.0 | 26574 | 0.4440 | 0.8673 |
|
153 |
+
| 0.1605 | 87.0 | 26883 | 0.4504 | 0.8731 |
|
154 |
+
| 0.1586 | 88.0 | 27192 | 0.4551 | 0.8667 |
|
155 |
+
| 0.1558 | 89.0 | 27501 | 0.4498 | 0.8670 |
|
156 |
+
| 0.1558 | 90.0 | 27810 | 0.4516 | 0.8718 |
|
157 |
+
| 0.1587 | 91.0 | 28119 | 0.4450 | 0.8725 |
|
158 |
+
| 0.1587 | 92.0 | 28428 | 0.4435 | 0.8706 |
|
159 |
+
| 0.1505 | 93.0 | 28737 | 0.4459 | 0.8722 |
|
160 |
+
| 0.1492 | 94.0 | 29046 | 0.4578 | 0.8673 |
|
161 |
+
| 0.1492 | 95.0 | 29355 | 0.4499 | 0.8725 |
|
162 |
+
| 0.1459 | 96.0 | 29664 | 0.4494 | 0.8703 |
|
163 |
+
| 0.1459 | 97.0 | 29973 | 0.4533 | 0.8697 |
|
164 |
+
| 0.1481 | 98.0 | 30282 | 0.4524 | 0.8652 |
|
165 |
+
| 0.1477 | 99.0 | 30591 | 0.4496 | 0.8715 |
|
166 |
+
| 0.1477 | 100.0 | 30900 | 0.4523 | 0.8661 |
|
167 |
+
|
168 |
+
|
169 |
+
### Framework versions
|
170 |
+
|
171 |
+
- Transformers 4.37.2
|
172 |
+
- Pytorch 2.3.0
|
173 |
+
- Datasets 2.15.0
|
174 |
+
- Tokenizers 0.15.1
|
config.json
ADDED
@@ -0,0 +1,285 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_name_or_path": "microsoft/resnet-101",
|
3 |
+
"architectures": [
|
4 |
+
"ResNetForImageClassification"
|
5 |
+
],
|
6 |
+
"depths": [
|
7 |
+
3,
|
8 |
+
4,
|
9 |
+
23,
|
10 |
+
3
|
11 |
+
],
|
12 |
+
"downsample_in_bottleneck": false,
|
13 |
+
"downsample_in_first_stage": false,
|
14 |
+
"embedding_size": 64,
|
15 |
+
"hidden_act": "relu",
|
16 |
+
"hidden_sizes": [
|
17 |
+
256,
|
18 |
+
512,
|
19 |
+
1024,
|
20 |
+
2048
|
21 |
+
],
|
22 |
+
"id2label": {
|
23 |
+
"0": "Afghan_hound",
|
24 |
+
"1": "African_hunting_dog",
|
25 |
+
"10": "Border_terrier",
|
26 |
+
"100": "kuvasz",
|
27 |
+
"101": "malamute",
|
28 |
+
"102": "malinois",
|
29 |
+
"103": "miniature_pinscher",
|
30 |
+
"104": "miniature_poodle",
|
31 |
+
"105": "miniature_schnauzer",
|
32 |
+
"106": "otterhound",
|
33 |
+
"107": "papillon",
|
34 |
+
"108": "pug",
|
35 |
+
"109": "redbone",
|
36 |
+
"11": "Boston_bull",
|
37 |
+
"110": "schipperke",
|
38 |
+
"111": "silky_terrier",
|
39 |
+
"112": "soft",
|
40 |
+
"113": "standard_poodle",
|
41 |
+
"114": "standard_schnauzer",
|
42 |
+
"115": "toy_poodle",
|
43 |
+
"116": "toy_terrier",
|
44 |
+
"117": "vizsla",
|
45 |
+
"118": "whippet",
|
46 |
+
"119": "wire",
|
47 |
+
"12": "Bouvier_des_Flandres",
|
48 |
+
"13": "Brabancon_griffon",
|
49 |
+
"14": "Brittany_spaniel",
|
50 |
+
"15": "Cardigan",
|
51 |
+
"16": "Chesapeake_Bay_retriever",
|
52 |
+
"17": "Chihuahua",
|
53 |
+
"18": "Dandie_Dinmont",
|
54 |
+
"19": "Doberman",
|
55 |
+
"2": "Airedale",
|
56 |
+
"20": "English_foxhound",
|
57 |
+
"21": "English_setter",
|
58 |
+
"22": "English_springer",
|
59 |
+
"23": "EntleBucher",
|
60 |
+
"24": "Eskimo_dog",
|
61 |
+
"25": "French_bulldog",
|
62 |
+
"26": "German_shepherd",
|
63 |
+
"27": "German_short",
|
64 |
+
"28": "Gordon_setter",
|
65 |
+
"29": "Great_Dane",
|
66 |
+
"3": "American_Staffordshire_terrier",
|
67 |
+
"30": "Great_Pyrenees",
|
68 |
+
"31": "Greater_Swiss_Mountain_dog",
|
69 |
+
"32": "Ibizan_hound",
|
70 |
+
"33": "Irish_setter",
|
71 |
+
"34": "Irish_terrier",
|
72 |
+
"35": "Irish_water_spaniel",
|
73 |
+
"36": "Irish_wolfhound",
|
74 |
+
"37": "Italian_greyhound",
|
75 |
+
"38": "Japanese_spaniel",
|
76 |
+
"39": "Kerry_blue_terrier",
|
77 |
+
"4": "Appenzeller",
|
78 |
+
"40": "Labrador_retriever",
|
79 |
+
"41": "Lakeland_terrier",
|
80 |
+
"42": "Leonberg",
|
81 |
+
"43": "Lhasa",
|
82 |
+
"44": "Maltese_dog",
|
83 |
+
"45": "Mexican_hairless",
|
84 |
+
"46": "Newfoundland",
|
85 |
+
"47": "Norfolk_terrier",
|
86 |
+
"48": "Norwegian_elkhound",
|
87 |
+
"49": "Norwich_terrier",
|
88 |
+
"5": "Australian_terrier",
|
89 |
+
"50": "Old_English_sheepdog",
|
90 |
+
"51": "Pekinese",
|
91 |
+
"52": "Pembroke",
|
92 |
+
"53": "Pomeranian",
|
93 |
+
"54": "Rhodesian_ridgeback",
|
94 |
+
"55": "Rottweiler",
|
95 |
+
"56": "Saint_Bernard",
|
96 |
+
"57": "Saluki",
|
97 |
+
"58": "Samoyed",
|
98 |
+
"59": "Scotch_terrier",
|
99 |
+
"6": "Bedlington_terrier",
|
100 |
+
"60": "Scottish_deerhound",
|
101 |
+
"61": "Sealyham_terrier",
|
102 |
+
"62": "Shetland_sheepdog",
|
103 |
+
"63": "Shih",
|
104 |
+
"64": "Siberian_husky",
|
105 |
+
"65": "Staffordshire_bullterrier",
|
106 |
+
"66": "Sussex_spaniel",
|
107 |
+
"67": "Tibetan_mastiff",
|
108 |
+
"68": "Tibetan_terrier",
|
109 |
+
"69": "Walker_hound",
|
110 |
+
"7": "Bernese_mountain_dog",
|
111 |
+
"70": "Weimaraner",
|
112 |
+
"71": "Welsh_springer_spaniel",
|
113 |
+
"72": "West_Highland_white_terrier",
|
114 |
+
"73": "Yorkshire_terrier",
|
115 |
+
"74": "affenpinscher",
|
116 |
+
"75": "basenji",
|
117 |
+
"76": "basset",
|
118 |
+
"77": "beagle",
|
119 |
+
"78": "black",
|
120 |
+
"79": "bloodhound",
|
121 |
+
"8": "Blenheim_spaniel",
|
122 |
+
"80": "bluetick",
|
123 |
+
"81": "borzoi",
|
124 |
+
"82": "boxer",
|
125 |
+
"83": "briard",
|
126 |
+
"84": "bull_mastiff",
|
127 |
+
"85": "cairn",
|
128 |
+
"86": "chow",
|
129 |
+
"87": "clumber",
|
130 |
+
"88": "cocker_spaniel",
|
131 |
+
"89": "collie",
|
132 |
+
"9": "Border_collie",
|
133 |
+
"90": "curly",
|
134 |
+
"91": "dhole",
|
135 |
+
"92": "dingo",
|
136 |
+
"93": "flat",
|
137 |
+
"94": "giant_schnauzer",
|
138 |
+
"95": "golden_retriever",
|
139 |
+
"96": "groenendael",
|
140 |
+
"97": "keeshond",
|
141 |
+
"98": "kelpie",
|
142 |
+
"99": "komondor"
|
143 |
+
},
|
144 |
+
"label2id": {
|
145 |
+
"Afghan_hound": "0",
|
146 |
+
"African_hunting_dog": "1",
|
147 |
+
"Airedale": "2",
|
148 |
+
"American_Staffordshire_terrier": "3",
|
149 |
+
"Appenzeller": "4",
|
150 |
+
"Australian_terrier": "5",
|
151 |
+
"Bedlington_terrier": "6",
|
152 |
+
"Bernese_mountain_dog": "7",
|
153 |
+
"Blenheim_spaniel": "8",
|
154 |
+
"Border_collie": "9",
|
155 |
+
"Border_terrier": "10",
|
156 |
+
"Boston_bull": "11",
|
157 |
+
"Bouvier_des_Flandres": "12",
|
158 |
+
"Brabancon_griffon": "13",
|
159 |
+
"Brittany_spaniel": "14",
|
160 |
+
"Cardigan": "15",
|
161 |
+
"Chesapeake_Bay_retriever": "16",
|
162 |
+
"Chihuahua": "17",
|
163 |
+
"Dandie_Dinmont": "18",
|
164 |
+
"Doberman": "19",
|
165 |
+
"English_foxhound": "20",
|
166 |
+
"English_setter": "21",
|
167 |
+
"English_springer": "22",
|
168 |
+
"EntleBucher": "23",
|
169 |
+
"Eskimo_dog": "24",
|
170 |
+
"French_bulldog": "25",
|
171 |
+
"German_shepherd": "26",
|
172 |
+
"German_short": "27",
|
173 |
+
"Gordon_setter": "28",
|
174 |
+
"Great_Dane": "29",
|
175 |
+
"Great_Pyrenees": "30",
|
176 |
+
"Greater_Swiss_Mountain_dog": "31",
|
177 |
+
"Ibizan_hound": "32",
|
178 |
+
"Irish_setter": "33",
|
179 |
+
"Irish_terrier": "34",
|
180 |
+
"Irish_water_spaniel": "35",
|
181 |
+
"Irish_wolfhound": "36",
|
182 |
+
"Italian_greyhound": "37",
|
183 |
+
"Japanese_spaniel": "38",
|
184 |
+
"Kerry_blue_terrier": "39",
|
185 |
+
"Labrador_retriever": "40",
|
186 |
+
"Lakeland_terrier": "41",
|
187 |
+
"Leonberg": "42",
|
188 |
+
"Lhasa": "43",
|
189 |
+
"Maltese_dog": "44",
|
190 |
+
"Mexican_hairless": "45",
|
191 |
+
"Newfoundland": "46",
|
192 |
+
"Norfolk_terrier": "47",
|
193 |
+
"Norwegian_elkhound": "48",
|
194 |
+
"Norwich_terrier": "49",
|
195 |
+
"Old_English_sheepdog": "50",
|
196 |
+
"Pekinese": "51",
|
197 |
+
"Pembroke": "52",
|
198 |
+
"Pomeranian": "53",
|
199 |
+
"Rhodesian_ridgeback": "54",
|
200 |
+
"Rottweiler": "55",
|
201 |
+
"Saint_Bernard": "56",
|
202 |
+
"Saluki": "57",
|
203 |
+
"Samoyed": "58",
|
204 |
+
"Scotch_terrier": "59",
|
205 |
+
"Scottish_deerhound": "60",
|
206 |
+
"Sealyham_terrier": "61",
|
207 |
+
"Shetland_sheepdog": "62",
|
208 |
+
"Shih": "63",
|
209 |
+
"Siberian_husky": "64",
|
210 |
+
"Staffordshire_bullterrier": "65",
|
211 |
+
"Sussex_spaniel": "66",
|
212 |
+
"Tibetan_mastiff": "67",
|
213 |
+
"Tibetan_terrier": "68",
|
214 |
+
"Walker_hound": "69",
|
215 |
+
"Weimaraner": "70",
|
216 |
+
"Welsh_springer_spaniel": "71",
|
217 |
+
"West_Highland_white_terrier": "72",
|
218 |
+
"Yorkshire_terrier": "73",
|
219 |
+
"affenpinscher": "74",
|
220 |
+
"basenji": "75",
|
221 |
+
"basset": "76",
|
222 |
+
"beagle": "77",
|
223 |
+
"black": "78",
|
224 |
+
"bloodhound": "79",
|
225 |
+
"bluetick": "80",
|
226 |
+
"borzoi": "81",
|
227 |
+
"boxer": "82",
|
228 |
+
"briard": "83",
|
229 |
+
"bull_mastiff": "84",
|
230 |
+
"cairn": "85",
|
231 |
+
"chow": "86",
|
232 |
+
"clumber": "87",
|
233 |
+
"cocker_spaniel": "88",
|
234 |
+
"collie": "89",
|
235 |
+
"curly": "90",
|
236 |
+
"dhole": "91",
|
237 |
+
"dingo": "92",
|
238 |
+
"flat": "93",
|
239 |
+
"giant_schnauzer": "94",
|
240 |
+
"golden_retriever": "95",
|
241 |
+
"groenendael": "96",
|
242 |
+
"keeshond": "97",
|
243 |
+
"kelpie": "98",
|
244 |
+
"komondor": "99",
|
245 |
+
"kuvasz": "100",
|
246 |
+
"malamute": "101",
|
247 |
+
"malinois": "102",
|
248 |
+
"miniature_pinscher": "103",
|
249 |
+
"miniature_poodle": "104",
|
250 |
+
"miniature_schnauzer": "105",
|
251 |
+
"otterhound": "106",
|
252 |
+
"papillon": "107",
|
253 |
+
"pug": "108",
|
254 |
+
"redbone": "109",
|
255 |
+
"schipperke": "110",
|
256 |
+
"silky_terrier": "111",
|
257 |
+
"soft": "112",
|
258 |
+
"standard_poodle": "113",
|
259 |
+
"standard_schnauzer": "114",
|
260 |
+
"toy_poodle": "115",
|
261 |
+
"toy_terrier": "116",
|
262 |
+
"vizsla": "117",
|
263 |
+
"whippet": "118",
|
264 |
+
"wire": "119"
|
265 |
+
},
|
266 |
+
"layer_type": "bottleneck",
|
267 |
+
"model_type": "resnet",
|
268 |
+
"num_channels": 3,
|
269 |
+
"out_features": [
|
270 |
+
"stage4"
|
271 |
+
],
|
272 |
+
"out_indices": [
|
273 |
+
4
|
274 |
+
],
|
275 |
+
"problem_type": "single_label_classification",
|
276 |
+
"stage_names": [
|
277 |
+
"stem",
|
278 |
+
"stage1",
|
279 |
+
"stage2",
|
280 |
+
"stage3",
|
281 |
+
"stage4"
|
282 |
+
],
|
283 |
+
"torch_dtype": "float32",
|
284 |
+
"transformers_version": "4.37.2"
|
285 |
+
}
|
model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:19d6315abec1d2ee3e156a14bda711b7ecf7c8a4e8c4a70ee126b98a80c6faf7
|
3 |
+
size 171488800
|
preprocessor_config.json
ADDED
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"crop_pct": 0.875,
|
3 |
+
"do_normalize": true,
|
4 |
+
"do_rescale": true,
|
5 |
+
"do_resize": true,
|
6 |
+
"image_mean": [
|
7 |
+
0.485,
|
8 |
+
0.456,
|
9 |
+
0.406
|
10 |
+
],
|
11 |
+
"image_processor_type": "ConvNextImageProcessor",
|
12 |
+
"image_std": [
|
13 |
+
0.229,
|
14 |
+
0.224,
|
15 |
+
0.225
|
16 |
+
],
|
17 |
+
"resample": 3,
|
18 |
+
"rescale_factor": 0.00392156862745098,
|
19 |
+
"size": {
|
20 |
+
"shortest_edge": 224
|
21 |
+
}
|
22 |
+
}
|
training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e589cc674be7319ce8277ff9c33b1121c56d3dd49e0dfdce0076b2f5374b7aed
|
3 |
+
size 4728
|