Unverified Commit ff5dc219 authored by Pavel Yakubovskiy's avatar Pavel Yakubovskiy Committed by GitHub

Merge pull request #48 from qubvel/fix-mobilenet

Fix mobilenet
parents 45ae8785 11c4e4e1
......@@ -11,7 +11,7 @@ matrix:
- python: 3.6
env: KERAS_VERSION=head
- python: 3.6
env: KERAS_VERSION=2.1.5
env: KERAS_VERSION=2.2.0
- python: 3.6
env: KERAS_VERSION=2.2.4
......
......@@ -3,6 +3,8 @@ from classification_models import resnext
from . import inception_resnet_v2 as irv2
from . import inception_v3 as iv3
from . import mobilenet as mbn
from . import mobilenetv2 as mbn2
# replace backbones with others, which have corrected padding mode in first pooling
Classifiers._models.update({
......@@ -10,6 +12,8 @@ Classifiers._models.update({
'inceptionv3': [iv3.InceptionV3, iv3.preprocess_input],
'resnext50': [resnext.ResNeXt50, resnext.models.preprocess_input],
'resnext101': [resnext.ResNeXt101, resnext.models.preprocess_input],
'mobilenet': [mbn.MobileNet, mbn.preprocess_input],
'mobilenetv2': [mbn2.MobileNetV2, mbn2.preprocess_input],
})
DEFAULT_FEATURE_LAYERS = {
......
This diff is collapsed.
This diff is collapsed.
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment