Add generic constructor

Add a parameterized constructor for creating any of the EfficientNet
variants, i.e. from EfficientNetB0 to EfficientNetB7.

Author: krikru

efficientnet/keras.py

@@ -3,6 +3,7 @@
 
 from .preprocessing import center_crop_and_resize
 
+EfficientNetBn = inject_keras_modules(model.EfficientNetBn)
 EfficientNetB0 = inject_keras_modules(model.EfficientNetB0)
 EfficientNetB1 = inject_keras_modules(model.EfficientNetB1)
 EfficientNetB2 = inject_keras_modules(model.EfficientNetB2)

efficientnet/model.py

@@ -473,6 +473,19 @@ def EfficientNet(width_coefficient,
     return model
 
 
+def EfficientNetBn(n, *args, **kwargs):
+    return [
+        EfficientNetB0,
+        EfficientNetB1,
+        EfficientNetB2,
+        EfficientNetB3,
+        EfficientNetB4,
+        EfficientNetB5,
+        EfficientNetB6,
+        EfficientNetB7,
+    ][n](*args, **kwargs)
+
+
 def EfficientNetB0(include_top=True,
                    weights='imagenet',
                    input_tensor=None,

efficientnet/tfkeras.py

@@ -3,6 +3,7 @@
 
 from .preprocessing import center_crop_and_resize
 
+EfficientNetBn = inject_tfkeras_modules(model.EfficientNetBn)
 EfficientNetB0 = inject_tfkeras_modules(model.EfficientNetB0)
 EfficientNetB1 = inject_tfkeras_modules(model.EfficientNetB1)
 EfficientNetB2 = inject_tfkeras_modules(model.EfficientNetB2)