|
|
@ -24,14 +24,12 @@ An implementation of EfficienNet that covers variety of related models with effi
|
|
|
|
|
|
|
|
|
|
|
|
Hacked together by Ross Wightman
|
|
|
|
Hacked together by Ross Wightman
|
|
|
|
"""
|
|
|
|
"""
|
|
|
|
|
|
|
|
from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD, IMAGENET_INCEPTION_MEAN, IMAGENET_INCEPTION_STD
|
|
|
|
from .efficientnet_builder import *
|
|
|
|
from .efficientnet_builder import *
|
|
|
|
from .feature_hooks import FeatureHooks
|
|
|
|
from .feature_hooks import FeatureHooks
|
|
|
|
from .registry import register_model
|
|
|
|
|
|
|
|
from .helpers import load_pretrained, adapt_model_from_file
|
|
|
|
from .helpers import load_pretrained, adapt_model_from_file
|
|
|
|
from .layers import SelectAdaptivePool2d
|
|
|
|
from .layers import SelectAdaptivePool2d
|
|
|
|
from timm.models.layers import create_conv2d
|
|
|
|
from .registry import register_model
|
|
|
|
from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD, IMAGENET_INCEPTION_MEAN, IMAGENET_INCEPTION_STD
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
__all__ = ['EfficientNet']
|
|
|
|
__all__ = ['EfficientNet']
|
|
|
|
|
|
|
|
|
|
|
@ -1207,8 +1205,6 @@ def efficientnet_b3_pruned(pretrained=False, **kwargs):
|
|
|
|
return model
|
|
|
|
return model
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@register_model
|
|
|
|
@register_model
|
|
|
|
def tf_efficientnet_b0(pretrained=False, **kwargs):
|
|
|
|
def tf_efficientnet_b0(pretrained=False, **kwargs):
|
|
|
|
""" EfficientNet-B0. Tensorflow compatible variant """
|
|
|
|
""" EfficientNet-B0. Tensorflow compatible variant """
|
|
|
|