From b6e1d6524359ace0f9b9050f7523674c26026a4c Mon Sep 17 00:00:00 2001 From: Pavel Yakubovskiy Date: Tue, 16 Jul 2019 17:50:00 +0300 Subject: [PATCH] Add b4-b5 encoders (#131) * Update __init__.py * Update requirements.txt --- requirements.txt | 2 +- segmentation_models/backbones/__init__.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/requirements.txt b/requirements.txt index b4442260..ad15fded 100644 --- 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 index d34cc8dd..147febea 100644 --- 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),