Skip to content

Commit 23709c0

Browse files
qubvelqubvel
committed
Rename with dot
1 parent fb85d02 commit 23709c0

File tree

14 files changed

+116
-116
lines changed

14 files changed

+116
-116
lines changed

‎segmentation_models_pytorch/encoders/densenet.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,7 @@ def load_state_dict(self, state_dict):
121121
},
122122
"pretrained_settings": {
123123
"imagenet": {
124-
"repo_id": "smp-hub/densenet121-imagenet",
124+
"repo_id": "smp-hub/densenet121.imagenet",
125125
"revision": "main",
126126
}
127127
},
@@ -136,7 +136,7 @@ def load_state_dict(self, state_dict):
136136
},
137137
"pretrained_settings": {
138138
"imagenet": {
139-
"repo_id": "smp-hub/densenet169-imagenet",
139+
"repo_id": "smp-hub/densenet169.imagenet",
140140
"revision": "main",
141141
}
142142
},
@@ -151,7 +151,7 @@ def load_state_dict(self, state_dict):
151151
},
152152
"pretrained_settings": {
153153
"imagenet": {
154-
"repo_id": "smp-hub/densenet201-imagenet",
154+
"repo_id": "smp-hub/densenet201.imagenet",
155155
"revision": "main",
156156
}
157157
},
@@ -166,7 +166,7 @@ def load_state_dict(self, state_dict):
166166
},
167167
"pretrained_settings": {
168168
"imagenet": {
169-
"repo_id": "smp-hub/densenet161-imagenet",
169+
"repo_id": "smp-hub/densenet161.imagenet",
170170
"revision": "main",
171171
}
172172
},

‎segmentation_models_pytorch/encoders/dpn.py‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ def load_state_dict(self, state_dict, **kwargs):
106106
"encoder": DPNEncoder,
107107
"pretrained_settings": {
108108
"imagenet": {
109-
"repo_id": "smp-hub/dpn68-imagenet",
109+
"repo_id": "smp-hub/dpn68.imagenet",
110110
"revision": "main",
111111
}
112112
},
@@ -127,7 +127,7 @@ def load_state_dict(self, state_dict, **kwargs):
127127
"encoder": DPNEncoder,
128128
"pretrained_settings": {
129129
"imagenet+5k": {
130-
"repo_id": "smp-hub/dpn68b-imagenet-5k",
130+
"repo_id": "smp-hub/dpn68b.imagenet-5k",
131131
"revision": "main",
132132
}
133133
},
@@ -149,7 +149,7 @@ def load_state_dict(self, state_dict, **kwargs):
149149
"encoder": DPNEncoder,
150150
"pretrained_settings": {
151151
"imagenet+5k": {
152-
"repo_id": "smp-hub/dpn92-imagenet-5k",
152+
"repo_id": "smp-hub/dpn92.imagenet-5k",
153153
"revision": "main",
154154
}
155155
},
@@ -169,7 +169,7 @@ def load_state_dict(self, state_dict, **kwargs):
169169
"encoder": DPNEncoder,
170170
"pretrained_settings": {
171171
"imagenet": {
172-
"repo_id": "smp-hub/dpn98-imagenet",
172+
"repo_id": "smp-hub/dpn98.imagenet",
173173
"revision": "main",
174174
}
175175
},
@@ -189,7 +189,7 @@ def load_state_dict(self, state_dict, **kwargs):
189189
"encoder": DPNEncoder,
190190
"pretrained_settings": {
191191
"imagenet+5k": {
192-
"repo_id": "smp-hub/dpn107-imagenet-5k",
192+
"repo_id": "smp-hub/dpn107.imagenet-5k",
193193
"revision": "main",
194194
}
195195
},
@@ -209,7 +209,7 @@ def load_state_dict(self, state_dict, **kwargs):
209209
"encoder": DPNEncoder,
210210
"pretrained_settings": {
211211
"imagenet": {
212-
"repo_id": "smp-hub/dpn131-imagenet",
212+
"repo_id": "smp-hub/dpn131.imagenet",
213213
"revision": "main",
214214
}
215215
},

‎segmentation_models_pytorch/encoders/efficientnet.py‎

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -114,11 +114,11 @@ def load_state_dict(self, state_dict, **kwargs):
114114
"encoder": EfficientNetEncoder,
115115
"pretrained_settings": {
116116
"imagenet": {
117-
"repo_id": "smp-hub/efficientnet-b0-imagenet",
117+
"repo_id": "smp-hub/efficientnet-b0.imagenet",
118118
"revision": "main",
119119
},
120120
"advprop": {
121-
"repo_id": "smp-hub/efficientnet-b0-advprop",
121+
"repo_id": "smp-hub/efficientnet-b0.advprop",
122122
"revision": "main",
123123
},
124124
},
@@ -132,11 +132,11 @@ def load_state_dict(self, state_dict, **kwargs):
132132
"encoder": EfficientNetEncoder,
133133
"pretrained_settings": {
134134
"imagenet": {
135-
"repo_id": "smp-hub/efficientnet-b1-imagenet",
135+
"repo_id": "smp-hub/efficientnet-b1.imagenet",
136136
"revision": "main",
137137
},
138138
"advprop": {
139-
"repo_id": "smp-hub/efficientnet-b1-advprop",
139+
"repo_id": "smp-hub/efficientnet-b1.advprop",
140140
"revision": "main",
141141
},
142142
},
@@ -150,11 +150,11 @@ def load_state_dict(self, state_dict, **kwargs):
150150
"encoder": EfficientNetEncoder,
151151
"pretrained_settings": {
152152
"imagenet": {
153-
"repo_id": "smp-hub/efficientnet-b2-imagenet",
153+
"repo_id": "smp-hub/efficientnet-b2.imagenet",
154154
"revision": "main",
155155
},
156156
"advprop": {
157-
"repo_id": "smp-hub/efficientnet-b2-advprop",
157+
"repo_id": "smp-hub/efficientnet-b2.advprop",
158158
"revision": "main",
159159
},
160160
},
@@ -168,11 +168,11 @@ def load_state_dict(self, state_dict, **kwargs):
168168
"encoder": EfficientNetEncoder,
169169
"pretrained_settings": {
170170
"imagenet": {
171-
"repo_id": "smp-hub/efficientnet-b3-imagenet",
171+
"repo_id": "smp-hub/efficientnet-b3.imagenet",
172172
"revision": "main",
173173
},
174174
"advprop": {
175-
"repo_id": "smp-hub/efficientnet-b3-advprop",
175+
"repo_id": "smp-hub/efficientnet-b3.advprop",
176176
"revision": "main",
177177
},
178178
},
@@ -186,11 +186,11 @@ def load_state_dict(self, state_dict, **kwargs):
186186
"encoder": EfficientNetEncoder,
187187
"pretrained_settings": {
188188
"imagenet": {
189-
"repo_id": "smp-hub/efficientnet-b4-imagenet",
189+
"repo_id": "smp-hub/efficientnet-b4.imagenet",
190190
"revision": "main",
191191
},
192192
"advprop": {
193-
"repo_id": "smp-hub/efficientnet-b4-advprop",
193+
"repo_id": "smp-hub/efficientnet-b4.advprop",
194194
"revision": "main",
195195
},
196196
},
@@ -204,11 +204,11 @@ def load_state_dict(self, state_dict, **kwargs):
204204
"encoder": EfficientNetEncoder,
205205
"pretrained_settings": {
206206
"imagenet": {
207-
"repo_id": "smp-hub/efficientnet-b5-imagenet",
207+
"repo_id": "smp-hub/efficientnet-b5.imagenet",
208208
"revision": "main",
209209
},
210210
"advprop": {
211-
"repo_id": "smp-hub/efficientnet-b5-advprop",
211+
"repo_id": "smp-hub/efficientnet-b5.advprop",
212212
"revision": "main",
213213
},
214214
},
@@ -222,11 +222,11 @@ def load_state_dict(self, state_dict, **kwargs):
222222
"encoder": EfficientNetEncoder,
223223
"pretrained_settings": {
224224
"imagenet": {
225-
"repo_id": "smp-hub/efficientnet-b6-imagenet",
225+
"repo_id": "smp-hub/efficientnet-b6.imagenet",
226226
"revision": "main",
227227
},
228228
"advprop": {
229-
"repo_id": "smp-hub/efficientnet-b6-advprop",
229+
"repo_id": "smp-hub/efficientnet-b6.advprop",
230230
"revision": "main",
231231
},
232232
},
@@ -240,11 +240,11 @@ def load_state_dict(self, state_dict, **kwargs):
240240
"encoder": EfficientNetEncoder,
241241
"pretrained_settings": {
242242
"imagenet": {
243-
"repo_id": "smp-hub/efficientnet-b7-imagenet",
243+
"repo_id": "smp-hub/efficientnet-b7.imagenet",
244244
"revision": "main",
245245
},
246246
"advprop": {
247-
"repo_id": "smp-hub/efficientnet-b7-advprop",
247+
"repo_id": "smp-hub/efficientnet-b7.advprop",
248248
"revision": "main",
249249
},
250250
},

‎segmentation_models_pytorch/encoders/inceptionresnetv2.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -118,11 +118,11 @@ def load_state_dict(self, state_dict, **kwargs):
118118
"encoder": InceptionResNetV2Encoder,
119119
"pretrained_settings": {
120120
"imagenet": {
121-
"repo_id": "smp-hub/inceptionresnetv2-imagenet",
121+
"repo_id": "smp-hub/inceptionresnetv2.imagenet",
122122
"revision": "main",
123123
},
124124
"imagenet+background": {
125-
"repo_id": "smp-hub/inceptionresnetv2-imagenet-background",
125+
"repo_id": "smp-hub/inceptionresnetv2.imagenet-background",
126126
"revision": "main",
127127
},
128128
},

‎segmentation_models_pytorch/encoders/inceptionv4.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -99,11 +99,11 @@ def load_state_dict(self, state_dict, **kwargs):
9999
"encoder": InceptionV4Encoder,
100100
"pretrained_settings": {
101101
"imagenet": {
102-
"repo_id": "smp-hub/inceptionv4-imagenet",
102+
"repo_id": "smp-hub/inceptionv4.imagenet",
103103
"revision": "main",
104104
},
105105
"imagenet+background": {
106-
"repo_id": "smp-hub/inceptionv4-imagenet-background",
106+
"repo_id": "smp-hub/inceptionv4.imagenet-background",
107107
"revision": "main",
108108
},
109109
},

‎segmentation_models_pytorch/encoders/mix_transformer.py‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -596,7 +596,7 @@ def load_state_dict(self, state_dict):
596596
"encoder": MixVisionTransformerEncoder,
597597
"pretrained_settings": {
598598
"imagenet": {
599-
"repo_id": "smp-hub/mit_b0-imagenet",
599+
"repo_id": "smp-hub/mit_b0.imagenet",
600600
"revision": "main",
601601
}
602602
},
@@ -618,7 +618,7 @@ def load_state_dict(self, state_dict):
618618
"encoder": MixVisionTransformerEncoder,
619619
"pretrained_settings": {
620620
"imagenet": {
621-
"repo_id": "smp-hub/mit_b1-imagenet",
621+
"repo_id": "smp-hub/mit_b1.imagenet",
622622
"revision": "main",
623623
}
624624
},
@@ -640,7 +640,7 @@ def load_state_dict(self, state_dict):
640640
"encoder": MixVisionTransformerEncoder,
641641
"pretrained_settings": {
642642
"imagenet": {
643-
"repo_id": "smp-hub/mit_b2-imagenet",
643+
"repo_id": "smp-hub/mit_b2.imagenet",
644644
"revision": "main",
645645
}
646646
},
@@ -662,7 +662,7 @@ def load_state_dict(self, state_dict):
662662
"encoder": MixVisionTransformerEncoder,
663663
"pretrained_settings": {
664664
"imagenet": {
665-
"repo_id": "smp-hub/mit_b3-imagenet",
665+
"repo_id": "smp-hub/mit_b3.imagenet",
666666
"revision": "main",
667667
}
668668
},
@@ -684,7 +684,7 @@ def load_state_dict(self, state_dict):
684684
"encoder": MixVisionTransformerEncoder,
685685
"pretrained_settings": {
686686
"imagenet": {
687-
"repo_id": "smp-hub/mit_b4-imagenet",
687+
"repo_id": "smp-hub/mit_b4.imagenet",
688688
"revision": "main",
689689
}
690690
},
@@ -706,7 +706,7 @@ def load_state_dict(self, state_dict):
706706
"encoder": MixVisionTransformerEncoder,
707707
"pretrained_settings": {
708708
"imagenet": {
709-
"repo_id": "smp-hub/mit_b5-imagenet",
709+
"repo_id": "smp-hub/mit_b5.imagenet",
710710
"revision": "main",
711711
}
712712
},

‎segmentation_models_pytorch/encoders/mobilenet.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ def load_state_dict(self, state_dict, **kwargs):
8585
"encoder": MobileNetV2Encoder,
8686
"pretrained_settings": {
8787
"imagenet": {
88-
"repo_id": "smp-hub/mobilenet_v2-imagenet",
88+
"repo_id": "smp-hub/mobilenet_v2.imagenet",
8989
"revision": "main",
9090
}
9191
},

‎segmentation_models_pytorch/encoders/mobileone.py‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -493,7 +493,7 @@ def reparameterize_model(model: torch.nn.Module) -> nn.Module:
493493
"encoder": MobileOne,
494494
"pretrained_settings": {
495495
"imagenet": {
496-
"repo_id": "smp-hub/mobileone_s0-imagenet",
496+
"repo_id": "smp-hub/mobileone_s0.imagenet",
497497
"revision": "main",
498498
}
499499
},
@@ -508,7 +508,7 @@ def reparameterize_model(model: torch.nn.Module) -> nn.Module:
508508
"encoder": MobileOne,
509509
"pretrained_settings": {
510510
"imagenet": {
511-
"repo_id": "smp-hub/mobileone_s1-imagenet",
511+
"repo_id": "smp-hub/mobileone_s1.imagenet",
512512
"revision": "main",
513513
}
514514
},
@@ -522,7 +522,7 @@ def reparameterize_model(model: torch.nn.Module) -> nn.Module:
522522
"encoder": MobileOne,
523523
"pretrained_settings": {
524524
"imagenet": {
525-
"repo_id": "smp-hub/mobileone_s2-imagenet",
525+
"repo_id": "smp-hub/mobileone_s2.imagenet",
526526
"revision": "main",
527527
}
528528
},
@@ -536,7 +536,7 @@ def reparameterize_model(model: torch.nn.Module) -> nn.Module:
536536
"encoder": MobileOne,
537537
"pretrained_settings": {
538538
"imagenet": {
539-
"repo_id": "smp-hub/mobileone_s3-imagenet",
539+
"repo_id": "smp-hub/mobileone_s3.imagenet",
540540
"revision": "main",
541541
}
542542
},
@@ -550,7 +550,7 @@ def reparameterize_model(model: torch.nn.Module) -> nn.Module:
550550
"encoder": MobileOne,
551551
"pretrained_settings": {
552552
"imagenet": {
553-
"repo_id": "smp-hub/mobileone_s4-imagenet",
553+
"repo_id": "smp-hub/mobileone_s4.imagenet",
554554
"revision": "main",
555555
}
556556
},

0 commit comments

Comments
 (0)