Unverified Commit 316769a9 authored by Pavel Yakubovskiy's avatar Pavel Yakubovskiy Committed by GitHub

Update __init__.py

parent ef635038
......@@ -22,8 +22,8 @@ Classifiers._models.update({
'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 @@ DEFAULT_FEATURE_LAYERS = {
'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),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment