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)