def build_rpn(cfg, in_channels):
"""
This gives the gist of it. Not super important because it doesn't change as much
"""
if cfg.MODEL.RETINANET_ON:
return build_retinanet(cfg, in_channels)
return RPNModule(cfg, in_channels)
def __init__(self, cfg):
super(RetinaNet, self).__init__()
self.cfg = copy.deepcopy(cfg)
self.backbone = build_backbone(cfg)
self.rpn = build_retinanet(cfg, self.backbone.out_channels)
self.mask = None
if cfg.MODEL.MASK_ON:
self.mask = build_roi_mask_head(cfg)
def build_rpn(cfg, in_channels):
"""
This gives the gist of it. Not super important because it doesn't change as much
"""
if cfg.MODEL.RETINANET_ON:
return build_retinanet(cfg, in_channels)
rpn = RPNModule(cfg, in_channels)
if not cfg.MODEL.TRAIN_RPN:
for p in rpn.parameters():
p.requires_grad = False
return rpn
def build_rpn(cfg, in_channels):
"""
This gives the gist of it. Not super important because it doesn't change as much
"""
# add by hui ###############################
if cfg.MODEL.LOC_ON:
return build_location_net(cfg, in_channels)
if cfg.MODEL.FCOS_ON:
if cfg.MODEL.FCOS.CASCADE_ON:
return build_cascade_fcos(cfg, in_channels)
else:
return build_fcos(cfg, in_channels)
if cfg.MODEL.GAU_ON:
return build_gaussian_net(cfg, in_channels)
############################################################
if cfg.MODEL.RETINANET_ON:
# add by hui ###############################
if cfg.FREEANCHOR.FREEANCHOR_ON:
return retinanet_fa.build_retinanet(cfg, in_channels)
############################################################
return build_retinanet(cfg, in_channels)
return RPNModule(cfg, in_channels)
def build_rpn(cfg, in_channels):
if cfg.MODEL.RETINANET_ON:
return build_retinanet(cfg, in_channels)
return RPNModule(cfg, in_channels)
