first commit

ssd/modeling/backbone/efficient_net/__init__.py

@@ -0,0 +1,14 @@
+from ssd.modeling import registry
+from .efficient_net import EfficientNet
+
+__all__ = ['efficient_net_b3', 'EfficientNet']
+
+
+@registry.BACKBONES.register('efficient_net-b3')
+def efficient_net_b3(cfg, pretrained=True):
+    if pretrained:
+        model = EfficientNet.from_pretrained('efficientnet-b3')
+    else:
+        model = EfficientNet.from_name('efficientnet-b3')
+
+    return model