Merge pull request #1192 from Hannah37/master

modified sequence of densenet
motoki-121115 · Dec 9, 2019 · 216e6de · 216e6de
2 parents 264816b + 7d0129c
commit 216e6de
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/keras_retinanet/models/__init__.py b/keras_retinanet/models/__init__.py
@@ -49,16 +49,16 @@ def preprocess_image(self, inputs):
 def backbone(backbone_name):
     """ Returns a backbone object for the given backbone.
     """
-    if 'seresnext' in backbone_name or 'seresnet' in backbone_name or 'senet' in backbone_name:
+    if 'densenet' in backbone_name:
+        from .densenet import DenseNetBackbone as b
+    elif 'seresnext' in backbone_name or 'seresnet' in backbone_name or 'senet' in backbone_name:
         from .senet import SeBackbone as b
     elif 'resnet' in backbone_name:
         from .resnet import ResNetBackbone as b
     elif 'mobilenet' in backbone_name:
         from .mobilenet import MobileNetBackbone as b
     elif 'vgg' in backbone_name:
-        from .vgg import VGGBackbone as b
-    elif 'densenet' in backbone_name:
-        from .densenet import DenseNetBackbone as b
+        from .vgg import VGGBackbone as b   
     elif 'EfficientNet' in backbone_name:
         from .effnet import EfficientNetBackbone as b
     else: