@@ -117,7 +117,14 @@ def _cfg(url='', **kwargs):
117117 url = 'https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/efficientnet_em_ra2-66250f76.pth' ,
118118 input_size = (3 , 240 , 240 ), pool_size = (8 , 8 ), crop_pct = 0.882 ),
119119 'efficientnet_el' : _cfg (
120- url = '' , input_size = (3 , 300 , 300 ), pool_size = (10 , 10 ), crop_pct = 0.904 ),
120+ url = 'https://github.com/DeGirum/pruned-models/releases/download/efficientnet_v1.0/efficientnet_el.pth' ,
121+ input_size = (3 , 300 , 300 ), pool_size = (10 , 10 ), crop_pct = 0.904 ),
122+
123+ 'efficientnet_es_pruned' : _cfg (
124+ url = 'https://github.com/DeGirum/pruned-models/releases/download/efficientnet_v1.0/efficientnet_es_pruned75.pth' ),
125+ 'efficientnet_el_pruned' : _cfg (
126+ url = 'https://github.com/DeGirum/pruned-models/releases/download/efficientnet_v1.0/efficientnet_el_pruned70.pth' ,
127+ input_size = (3 , 300 , 300 ), pool_size = (10 , 10 ), crop_pct = 0.904 ),
121128
122129 'efficientnet_cc_b0_4e' : _cfg (url = '' ),
123130 'efficientnet_cc_b0_8e' : _cfg (url = '' ),
@@ -1113,6 +1120,12 @@ def efficientnet_es(pretrained=False, **kwargs):
11131120 'efficientnet_es' , channel_multiplier = 1.0 , depth_multiplier = 1.0 , pretrained = pretrained , ** kwargs )
11141121 return model
11151122
1123+ @register_model
1124+ def efficientnet_es_pruned (pretrained = False , ** kwargs ):
1125+ """ EfficientNet-Edge Small Pruned. For more info: https://github.com/DeGirum/pruned-models/releases/tag/efficientnet_v1.0"""
1126+ model = _gen_efficientnet_edge (
1127+ 'efficientnet_es_pruned' , channel_multiplier = 1.0 , depth_multiplier = 1.0 , pretrained = pretrained , ** kwargs )
1128+ return model
11161129
11171130@register_model
11181131def efficientnet_em (pretrained = False , ** kwargs ):
@@ -1129,6 +1142,12 @@ def efficientnet_el(pretrained=False, **kwargs):
11291142 'efficientnet_el' , channel_multiplier = 1.2 , depth_multiplier = 1.4 , pretrained = pretrained , ** kwargs )
11301143 return model
11311144
1145+ @register_model
1146+ def efficientnet_el_pruned (pretrained = False , ** kwargs ):
1147+ """ EfficientNet-Edge-Large pruned. For more info: https://github.com/DeGirum/pruned-models/releases/tag/efficientnet_v1.0"""
1148+ model = _gen_efficientnet_edge (
1149+ 'efficientnet_el_pruned' , channel_multiplier = 1.2 , depth_multiplier = 1.4 , pretrained = pretrained , ** kwargs )
1150+ return model
11321151
11331152@register_model
11341153def efficientnet_cc_b0_4e (pretrained = False , ** kwargs ):
0 commit comments