File tree Expand file tree Collapse file tree 2 files changed +7
-2
lines changed
keras_hub/src/models/mobilenet Expand file tree Collapse file tree 2 files changed +7
-2
lines changed Original file line number Diff line number Diff line change
1
+ from keras_hub .src .models .mobilenet .mobilenet_backbone import MobileNetBackbone
2
+ from keras_hub .src .models .mobilenet .mobilenet_presets import backbone_presets
3
+ from keras_hub .src .utils .preset_utils import register_presets
4
+
5
+ register_presets (backbone_presets , MobileNetBackbone )
Original file line number Diff line number Diff line change 1
1
"""MobileNet preset configurations."""
2
2
3
3
backbone_presets = {
4
- "mobilenetv3_small_050 " : {
4
+ "mobilenet_v3_small_050_imagenet " : {
5
5
"metadata" : {
6
6
"description" : (
7
- "Small Mobilenet V3 model pre-trained on the ImageNet 1k "
7
+ "Small MobileNet V3 model pre-trained on the ImageNet 1k "
8
8
"dataset at a 224x224 resolution."
9
9
),
10
10
"params" : 278784 ,
You can’t perform that action at this time.
0 commit comments