def __init__(self, cfgs, is_training):
super(DetectionNetworkR2CNN, self).__init__(cfgs, is_training)
self.anchor_sampler_r2cnn = AnchorSamplerR2CNN(cfgs)
self.proposal_sampler_r2cnn = ProposalSamplerR2CNN(cfgs)
self.losses = Loss(cfgs)
self.roi_extractor = RoIExtractor(cfgs)
self.box_head = BoxHead(cfgs)
def __init__(self, cfgs, is_training):
super(DetectionNetworkRefineRetinaNet,
self).__init__(cfgs, is_training)
self.anchor_sampler_retinenet = AnchorSamplerRetinaNet(cfgs)
self.refine_anchor_sampler_r3det = RefineAnchorSamplerR3Det(cfgs)
self.losses = Loss(self.cfgs)
def __init__(self, cfgs, is_training):
super(DetectionNetwork, self).__init__(cfgs, is_training)
self.anchor_sampler_csl = AnchorSamplerCSL(cfgs)
self.losses = Loss(self.cfgs)
self.coding_len = cfgs.ANGLE_RANGE // cfgs.OMEGA