Add b4-b5 encoders (#131)

* Update __init__.py * Update requirements.txt
qubvel · Jul 16, 2019 · b6e1d65 · b6e1d65
1 parent ef63503
commit b6e1d65
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -2,4 +2,4 @@ keras>=2.2.0
 keras_applications==1.0.7
 scikit-image
 image-classifiers==0.2.0
-efficientnet>=0.0.3
+efficientnet>=0.0.4
diff --git a/segmentation_models/backbones/__init__.py b/segmentation_models/backbones/__init__.py
@@ -22,8 +22,8 @@
     'efficientnetb3': [eff.EfficientNetB3, eff.preprocess_input],
 
     # weights are not released
-#     'efficientnetb4': [eff.EfficientNetB4, eff.preprocess_input],
-#     'efficientnetb5': [eff.EfficientNetB5, eff.preprocess_input],
+    'efficientnetb4': [eff.EfficientNetB4, eff.preprocess_input],
+    'efficientnetb5': [eff.EfficientNetB5, eff.preprocess_input],
 #     'efficientnetb6': [eff.EfficientNetB6, eff.preprocess_input],
 #     'efficientnetb7': [eff.EfficientNetB7, eff.preprocess_input],
 
@@ -80,8 +80,8 @@
     'efficientnetb3': (278, 122, 76, 30),
 
     # weights are not released
-#     'efficientnetb4': (342, 154, 92, 30),
-#     'efficientnetb5': (419, 199, 121, 43),
+    'efficientnetb4': (342, 154, 92, 30),
+    'efficientnetb5': (419, 199, 121, 43),
 #     'efficientnetb6': (483, 231, 137, 43),
 #     'efficientnetb7': (592, 276, 166, 56),