rexologue commited on
Commit
fe948e7
·
verified ·
1 Parent(s): 0ba648b

Add TIMM checkpoint

Browse files
Files changed (4) hide show
  1. README.md +126 -3
  2. config.json +196 -0
  3. labels.json +94 -0
  4. pytorch_model.bin +3 -0
README.md CHANGED
@@ -1,3 +1,126 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - timm
4
+ - vision
5
+ library_name: timm
6
+ ---
7
+
8
+ # rexologue/vit_large_384_for_trees
9
+
10
+ This repository hosts a fine-tuned `vit_large_patch16_384` classifier exported from the
11
+ [`vit_tune`](https://github.com/) training pipeline.
12
+
13
+ ## Labels
14
+
15
+ - abies_sibirica
16
+ - acer_campestre
17
+ - acer_ginnala
18
+ - acer_negundo
19
+ - acer_platanoides
20
+ - acer_pseudoplatanus
21
+ - acer_tataricum
22
+ - aesculus_hippocastanum
23
+ - alnus_alnobetula_fruticosa
24
+ - alnus_glutinosa
25
+ - alnus_incana
26
+ - arctostaphylos_uva-ursi
27
+ - berberis_vulgaris
28
+ - betula_nana
29
+ - betula_pendula
30
+ - betula_pubescens
31
+ - calluna_vulgaris
32
+ - cornus_alba
33
+ - cornus_mas
34
+ - cornus_sanguinea
35
+ - cornus_suecica
36
+ - cotoneaster_lucidus
37
+ - cotoneaster_melanocarpus
38
+ - daphne_mezereum
39
+ - elaeagnus_angustifolia
40
+ - euonymus_europaeus
41
+ - euonymus_verrucosus
42
+ - fraxinus_excelsior
43
+ - fraxinus_pennsylvanica
44
+ - genista_tinctoria
45
+ - hippophae_rhamnoides
46
+ - hypericum_maculatum
47
+ - hypericum_perforatum
48
+ - juglans_mandshurica
49
+ - juniperus_communis
50
+ - larix_sibirica
51
+ - ligustrum_vulgare
52
+ - lonicera_caerulea
53
+ - lonicera_nigra
54
+ - lonicera_tatarica
55
+ - lonicera_xylosteum
56
+ - physocarpus_opulifolius
57
+ - picea_abies
58
+ - picea_obovata
59
+ - pinus_sibirica
60
+ - pinus_sylvestris
61
+ - populus
62
+ - populus_alba
63
+ - populus_nigra
64
+ - populus_tremula
65
+ - potentilla_argentea
66
+ - potentilla_erecta
67
+ - potentilla_intermedia
68
+ - potentilla_norvegica
69
+ - potentilla_paradoxa
70
+ - potentilla_reptans
71
+ - potentilla_supina
72
+ - quercus_robur
73
+ - ribes_nigrum
74
+ - ribes_rubrum
75
+ - ribes_uva-crispa
76
+ - rosa_acicularis
77
+ - rosa_majalis
78
+ - rosa_rugosa
79
+ - rubus_arcticus
80
+ - rubus_caesius
81
+ - rubus_chamaemorus
82
+ - rubus_idaeus
83
+ - rubus_nessensis
84
+ - rubus_saxatilis
85
+ - salix_alba
86
+ - salix_caprea
87
+ - salix_cinerea
88
+ - salix_gmelinii
89
+ - salix_myrsinifolia
90
+ - salix_pentandra
91
+ - salix_triandra
92
+ - salix_viminalis
93
+ - sorbaria_sorbifolia
94
+ - sorbus_aucuparia
95
+ - spiraea_salicifolia
96
+ - symphoricarpos_albus
97
+ - tilia_cordata
98
+ - ulmus_glabra
99
+ - ulmus_laevis
100
+ - ulmus_pumila
101
+ - vaccinium_myrtillus
102
+ - vaccinium_oxycoccos
103
+ - vaccinium_uliginosum
104
+ - vaccinium_vitis-idaea
105
+ - viburnum_lantana
106
+ - viburnum_opulus
107
+
108
+ ## Usage
109
+
110
+ ```python
111
+ import timm
112
+ import torch
113
+
114
+ model = timm.create_model(
115
+ "vit_large_patch16_384",
116
+ num_classes=92,
117
+ pretrained=False,
118
+ )
119
+ state_dict = torch.hub.load_state_dict_from_url(
120
+ "https://huggingface.co/rexologue/vit_large_384_for_trees/resolve/main/pytorch_model.bin",
121
+ map_location="cpu",
122
+ file_name="rexologue--vit_large_384_for_trees.bin",
123
+ )
124
+ model.load_state_dict(state_dict)
125
+ model.eval()
126
+ ```
config.json ADDED
@@ -0,0 +1,196 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "vit_large_patch16_384"
4
+ ],
5
+ "model_type": "timm",
6
+ "timm_model_name": "vit_large_patch16_384",
7
+ "num_labels": 92,
8
+ "id2label": {
9
+ "0": "abies_sibirica",
10
+ "1": "acer_campestre",
11
+ "2": "acer_ginnala",
12
+ "3": "acer_negundo",
13
+ "4": "acer_platanoides",
14
+ "5": "acer_pseudoplatanus",
15
+ "6": "acer_tataricum",
16
+ "7": "aesculus_hippocastanum",
17
+ "8": "alnus_alnobetula_fruticosa",
18
+ "9": "alnus_glutinosa",
19
+ "10": "alnus_incana",
20
+ "11": "arctostaphylos_uva-ursi",
21
+ "12": "berberis_vulgaris",
22
+ "13": "betula_nana",
23
+ "14": "betula_pendula",
24
+ "15": "betula_pubescens",
25
+ "16": "calluna_vulgaris",
26
+ "17": "cornus_alba",
27
+ "18": "cornus_mas",
28
+ "19": "cornus_sanguinea",
29
+ "20": "cornus_suecica",
30
+ "21": "cotoneaster_lucidus",
31
+ "22": "cotoneaster_melanocarpus",
32
+ "23": "daphne_mezereum",
33
+ "24": "elaeagnus_angustifolia",
34
+ "25": "euonymus_europaeus",
35
+ "26": "euonymus_verrucosus",
36
+ "27": "fraxinus_excelsior",
37
+ "28": "fraxinus_pennsylvanica",
38
+ "29": "genista_tinctoria",
39
+ "30": "hippophae_rhamnoides",
40
+ "31": "hypericum_maculatum",
41
+ "32": "hypericum_perforatum",
42
+ "33": "juglans_mandshurica",
43
+ "34": "juniperus_communis",
44
+ "35": "larix_sibirica",
45
+ "36": "ligustrum_vulgare",
46
+ "37": "lonicera_caerulea",
47
+ "38": "lonicera_nigra",
48
+ "39": "lonicera_tatarica",
49
+ "40": "lonicera_xylosteum",
50
+ "41": "physocarpus_opulifolius",
51
+ "42": "picea_abies",
52
+ "43": "picea_obovata",
53
+ "44": "pinus_sibirica",
54
+ "45": "pinus_sylvestris",
55
+ "46": "populus",
56
+ "47": "populus_alba",
57
+ "48": "populus_nigra",
58
+ "49": "populus_tremula",
59
+ "50": "potentilla_argentea",
60
+ "51": "potentilla_erecta",
61
+ "52": "potentilla_intermedia",
62
+ "53": "potentilla_norvegica",
63
+ "54": "potentilla_paradoxa",
64
+ "55": "potentilla_reptans",
65
+ "56": "potentilla_supina",
66
+ "57": "quercus_robur",
67
+ "58": "ribes_nigrum",
68
+ "59": "ribes_rubrum",
69
+ "60": "ribes_uva-crispa",
70
+ "61": "rosa_acicularis",
71
+ "62": "rosa_majalis",
72
+ "63": "rosa_rugosa",
73
+ "64": "rubus_arcticus",
74
+ "65": "rubus_caesius",
75
+ "66": "rubus_chamaemorus",
76
+ "67": "rubus_idaeus",
77
+ "68": "rubus_nessensis",
78
+ "69": "rubus_saxatilis",
79
+ "70": "salix_alba",
80
+ "71": "salix_caprea",
81
+ "72": "salix_cinerea",
82
+ "73": "salix_gmelinii",
83
+ "74": "salix_myrsinifolia",
84
+ "75": "salix_pentandra",
85
+ "76": "salix_triandra",
86
+ "77": "salix_viminalis",
87
+ "78": "sorbaria_sorbifolia",
88
+ "79": "sorbus_aucuparia",
89
+ "80": "spiraea_salicifolia",
90
+ "81": "symphoricarpos_albus",
91
+ "82": "tilia_cordata",
92
+ "83": "ulmus_glabra",
93
+ "84": "ulmus_laevis",
94
+ "85": "ulmus_pumila",
95
+ "86": "vaccinium_myrtillus",
96
+ "87": "vaccinium_oxycoccos",
97
+ "88": "vaccinium_uliginosum",
98
+ "89": "vaccinium_vitis-idaea",
99
+ "90": "viburnum_lantana",
100
+ "91": "viburnum_opulus"
101
+ },
102
+ "label2id": {
103
+ "abies_sibirica": 0,
104
+ "acer_campestre": 1,
105
+ "acer_ginnala": 2,
106
+ "acer_negundo": 3,
107
+ "acer_platanoides": 4,
108
+ "acer_pseudoplatanus": 5,
109
+ "acer_tataricum": 6,
110
+ "aesculus_hippocastanum": 7,
111
+ "alnus_alnobetula_fruticosa": 8,
112
+ "alnus_glutinosa": 9,
113
+ "alnus_incana": 10,
114
+ "arctostaphylos_uva-ursi": 11,
115
+ "berberis_vulgaris": 12,
116
+ "betula_nana": 13,
117
+ "betula_pendula": 14,
118
+ "betula_pubescens": 15,
119
+ "calluna_vulgaris": 16,
120
+ "cornus_alba": 17,
121
+ "cornus_mas": 18,
122
+ "cornus_sanguinea": 19,
123
+ "cornus_suecica": 20,
124
+ "cotoneaster_lucidus": 21,
125
+ "cotoneaster_melanocarpus": 22,
126
+ "daphne_mezereum": 23,
127
+ "elaeagnus_angustifolia": 24,
128
+ "euonymus_europaeus": 25,
129
+ "euonymus_verrucosus": 26,
130
+ "fraxinus_excelsior": 27,
131
+ "fraxinus_pennsylvanica": 28,
132
+ "genista_tinctoria": 29,
133
+ "hippophae_rhamnoides": 30,
134
+ "hypericum_maculatum": 31,
135
+ "hypericum_perforatum": 32,
136
+ "juglans_mandshurica": 33,
137
+ "juniperus_communis": 34,
138
+ "larix_sibirica": 35,
139
+ "ligustrum_vulgare": 36,
140
+ "lonicera_caerulea": 37,
141
+ "lonicera_nigra": 38,
142
+ "lonicera_tatarica": 39,
143
+ "lonicera_xylosteum": 40,
144
+ "physocarpus_opulifolius": 41,
145
+ "picea_abies": 42,
146
+ "picea_obovata": 43,
147
+ "pinus_sibirica": 44,
148
+ "pinus_sylvestris": 45,
149
+ "populus": 46,
150
+ "populus_alba": 47,
151
+ "populus_nigra": 48,
152
+ "populus_tremula": 49,
153
+ "potentilla_argentea": 50,
154
+ "potentilla_erecta": 51,
155
+ "potentilla_intermedia": 52,
156
+ "potentilla_norvegica": 53,
157
+ "potentilla_paradoxa": 54,
158
+ "potentilla_reptans": 55,
159
+ "potentilla_supina": 56,
160
+ "quercus_robur": 57,
161
+ "ribes_nigrum": 58,
162
+ "ribes_rubrum": 59,
163
+ "ribes_uva-crispa": 60,
164
+ "rosa_acicularis": 61,
165
+ "rosa_majalis": 62,
166
+ "rosa_rugosa": 63,
167
+ "rubus_arcticus": 64,
168
+ "rubus_caesius": 65,
169
+ "rubus_chamaemorus": 66,
170
+ "rubus_idaeus": 67,
171
+ "rubus_nessensis": 68,
172
+ "rubus_saxatilis": 69,
173
+ "salix_alba": 70,
174
+ "salix_caprea": 71,
175
+ "salix_cinerea": 72,
176
+ "salix_gmelinii": 73,
177
+ "salix_myrsinifolia": 74,
178
+ "salix_pentandra": 75,
179
+ "salix_triandra": 76,
180
+ "salix_viminalis": 77,
181
+ "sorbaria_sorbifolia": 78,
182
+ "sorbus_aucuparia": 79,
183
+ "spiraea_salicifolia": 80,
184
+ "symphoricarpos_albus": 81,
185
+ "tilia_cordata": 82,
186
+ "ulmus_glabra": 83,
187
+ "ulmus_laevis": 84,
188
+ "ulmus_pumila": 85,
189
+ "vaccinium_myrtillus": 86,
190
+ "vaccinium_oxycoccos": 87,
191
+ "vaccinium_uliginosum": 88,
192
+ "vaccinium_vitis-idaea": 89,
193
+ "viburnum_lantana": 90,
194
+ "viburnum_opulus": 91
195
+ }
196
+ }
labels.json ADDED
@@ -0,0 +1,94 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "0": "abies_sibirica",
3
+ "1": "acer_campestre",
4
+ "2": "acer_ginnala",
5
+ "3": "acer_negundo",
6
+ "4": "acer_platanoides",
7
+ "5": "acer_pseudoplatanus",
8
+ "6": "acer_tataricum",
9
+ "7": "aesculus_hippocastanum",
10
+ "8": "alnus_alnobetula_fruticosa",
11
+ "9": "alnus_glutinosa",
12
+ "10": "alnus_incana",
13
+ "11": "arctostaphylos_uva-ursi",
14
+ "12": "berberis_vulgaris",
15
+ "13": "betula_nana",
16
+ "14": "betula_pendula",
17
+ "15": "betula_pubescens",
18
+ "16": "calluna_vulgaris",
19
+ "17": "cornus_alba",
20
+ "18": "cornus_mas",
21
+ "19": "cornus_sanguinea",
22
+ "20": "cornus_suecica",
23
+ "21": "cotoneaster_lucidus",
24
+ "22": "cotoneaster_melanocarpus",
25
+ "23": "daphne_mezereum",
26
+ "24": "elaeagnus_angustifolia",
27
+ "25": "euonymus_europaeus",
28
+ "26": "euonymus_verrucosus",
29
+ "27": "fraxinus_excelsior",
30
+ "28": "fraxinus_pennsylvanica",
31
+ "29": "genista_tinctoria",
32
+ "30": "hippophae_rhamnoides",
33
+ "31": "hypericum_maculatum",
34
+ "32": "hypericum_perforatum",
35
+ "33": "juglans_mandshurica",
36
+ "34": "juniperus_communis",
37
+ "35": "larix_sibirica",
38
+ "36": "ligustrum_vulgare",
39
+ "37": "lonicera_caerulea",
40
+ "38": "lonicera_nigra",
41
+ "39": "lonicera_tatarica",
42
+ "40": "lonicera_xylosteum",
43
+ "41": "physocarpus_opulifolius",
44
+ "42": "picea_abies",
45
+ "43": "picea_obovata",
46
+ "44": "pinus_sibirica",
47
+ "45": "pinus_sylvestris",
48
+ "46": "populus",
49
+ "47": "populus_alba",
50
+ "48": "populus_nigra",
51
+ "49": "populus_tremula",
52
+ "50": "potentilla_argentea",
53
+ "51": "potentilla_erecta",
54
+ "52": "potentilla_intermedia",
55
+ "53": "potentilla_norvegica",
56
+ "54": "potentilla_paradoxa",
57
+ "55": "potentilla_reptans",
58
+ "56": "potentilla_supina",
59
+ "57": "quercus_robur",
60
+ "58": "ribes_nigrum",
61
+ "59": "ribes_rubrum",
62
+ "60": "ribes_uva-crispa",
63
+ "61": "rosa_acicularis",
64
+ "62": "rosa_majalis",
65
+ "63": "rosa_rugosa",
66
+ "64": "rubus_arcticus",
67
+ "65": "rubus_caesius",
68
+ "66": "rubus_chamaemorus",
69
+ "67": "rubus_idaeus",
70
+ "68": "rubus_nessensis",
71
+ "69": "rubus_saxatilis",
72
+ "70": "salix_alba",
73
+ "71": "salix_caprea",
74
+ "72": "salix_cinerea",
75
+ "73": "salix_gmelinii",
76
+ "74": "salix_myrsinifolia",
77
+ "75": "salix_pentandra",
78
+ "76": "salix_triandra",
79
+ "77": "salix_viminalis",
80
+ "78": "sorbaria_sorbifolia",
81
+ "79": "sorbus_aucuparia",
82
+ "80": "spiraea_salicifolia",
83
+ "81": "symphoricarpos_albus",
84
+ "82": "tilia_cordata",
85
+ "83": "ulmus_glabra",
86
+ "84": "ulmus_laevis",
87
+ "85": "ulmus_pumila",
88
+ "86": "vaccinium_myrtillus",
89
+ "87": "vaccinium_oxycoccos",
90
+ "88": "vaccinium_uliginosum",
91
+ "89": "vaccinium_vitis-idaea",
92
+ "90": "viburnum_lantana",
93
+ "91": "viburnum_opulus"
94
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fbf2ffca17dd4940da79e06bb56f93c20ed9d9095582b8d62119252cce32fd72
3
+ size 1215252931