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
|