Skip to content

Commit b6e1d65

Browse files
authored
Add b4-b5 encoders (#131)
* Update __init__.py * Update requirements.txt
1 parent ef63503 commit b6e1d65

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,4 @@ keras>=2.2.0
22
keras_applications==1.0.7
33
scikit-image
44
image-classifiers==0.2.0
5-
efficientnet>=0.0.3
5+
efficientnet>=0.0.4

segmentation_models/backbones/__init__.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,8 @@
2222
'efficientnetb3': [eff.EfficientNetB3, eff.preprocess_input],
2323

2424
# weights are not released
25-
# 'efficientnetb4': [eff.EfficientNetB4, eff.preprocess_input],
26-
# 'efficientnetb5': [eff.EfficientNetB5, eff.preprocess_input],
25+
'efficientnetb4': [eff.EfficientNetB4, eff.preprocess_input],
26+
'efficientnetb5': [eff.EfficientNetB5, eff.preprocess_input],
2727
# 'efficientnetb6': [eff.EfficientNetB6, eff.preprocess_input],
2828
# 'efficientnetb7': [eff.EfficientNetB7, eff.preprocess_input],
2929

@@ -80,8 +80,8 @@
8080
'efficientnetb3': (278, 122, 76, 30),
8181

8282
# weights are not released
83-
# 'efficientnetb4': (342, 154, 92, 30),
84-
# 'efficientnetb5': (419, 199, 121, 43),
83+
'efficientnetb4': (342, 154, 92, 30),
84+
'efficientnetb5': (419, 199, 121, 43),
8585
# 'efficientnetb6': (483, 231, 137, 43),
8686
# 'efficientnetb7': (592, 276, 166, 56),
8787

0 commit comments

Comments
 (0)