def get_symbol_train(network, data_shape, alpha_bb8, **kwargs):
"""Wrapper for get symbol for train
Parameters
----------
network : str
name for the base network symbol
data_shape : int
input shape
kwargs : dict
see symbol_builder.get_symbol_train for more details
"""
if network.startswith('legacy'):
logging.warn('Using legacy model.')
return symbol_builder.import_module(network).get_symbol_train(**kwargs)
config = get_config(network, data_shape, **kwargs).copy()
config.update(kwargs)
networks = {
'vgg16': symbol_builder.get_vgg_reduced_fpn_train,
'resnet50': symbol_builder.get_resnet_fpn_train,
'resnet50d': symbol_builder.get_resnetd_fpn_train,
'resnet50deeplabv2': symbol_builder.get_resnetdeeplabv2_fpn_train,
'resnet50m': symbol_builder.get_resnetm_fpn_train,
'resnet50md': symbol_builder.get_resnetmd_fpn_train,
'resnet101': symbol_builder.get_resnet_fpn_train
}
if network not in networks:
raise ValueError("network {} not supported".format(network))
return networks[network](alpha_bb8=alpha_bb8, **config)
def get_symbol_split(network, data_shape, **kwargs):
"""Wrapper for get symbol for test
Parameters
----------
network : str
name for the base network symbol
data_shape : int
input shape
kwargs : dict
see symbol_builder.get_symbol for more details
"""
if network.startswith('legacy'):
logging.warn('Using legacy model.')
return symbol_builder.import_module(network).get_symbol_split(**kwargs)
config = get_config(network, data_shape, **kwargs).copy()
config.update(kwargs)
return symbol_builder.get_symbol_split(**config)
def get_symbol(network, data_shape, **kwargs):
"""Wrapper for get symbol for test
Parameters
----------
network : str
name for the base network symbol
data_shape : int
input shape
kwargs : dict
see symbol_builder.get_symbol for more details
"""
if network.startswith('legacy'):
logging.warn('Using legacy model.')
return symbol_builder.import_module(network).get_symbol(**kwargs)
config = get_config(network, data_shape, **kwargs).copy()
config.update(kwargs)
return symbol_builder.get_symbol(**config)
def get_symbol_train(network, data_shape, alpha_bb8, alpha_loc, use_dilated,
use_focalloss, **kwargs):
"""Wrapper for get symbol for train
Parameters
----------
network : str
name for the base network symbol
data_shape : int
input shape
kwargs : dict
see symbol_builder.get_symbol_train for more details
"""
if network.startswith('legacy'):
logging.warn('Using legacy model.')
return symbol_builder.import_module(network).get_symbol_train(**kwargs)
config = get_config(network, data_shape, **kwargs).copy()
config.update(kwargs)
return symbol_builder.get_symbol_train(alpha_bb8=alpha_bb8,
alpha_loc=alpha_loc,
use_dilated=use_dilated,
use_focalloss=use_focalloss,
**config)
