MODNet/modnet/models/backbones/__init__.py

11 lines
277 B
Python

from .wrapper import *
#------------------------------------------------------------------------------
# Replaceable Backbones
#------------------------------------------------------------------------------
SUPPORTED_BACKBONES = {
'mobilenetv2': MobileNetV2Backbone,
}