Add generic constructor

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

Author: krikru

efficientnet/model.py

@@ -42,6 +42,7 @@
 
 __all__ = [
     "EfficientNet",
+    "EfficientNetBn",
     "EfficientNetB0",
     "EfficientNetB1",
     "EfficientNetB2",
@@ -388,6 +389,20 @@ def _get_model_by_name(
     return model
 
 
+def EfficientNetBn(
+    n, include_top=True, input_shape=None, input_tensor=None, weights=None, classes=1000, pooling=None
+):
+    return _get_model_by_name(
+        "efficientnet-b{}".format(n),
+        include_top=include_top,
+        input_shape=input_shape,
+        input_tensor=input_tensor,
+        weights=weights,
+        classes=classes,
+        pooling=pooling,
+    )
+
+
 def EfficientNetB0(
     include_top=True, input_shape=None, input_tensor=None, weights=None, classes=1000, pooling=None
 ):