def __init__(self, cfgs, is_training):
super(DetectionNetworkR3DetDCL, self).__init__(cfgs, is_training)
self.anchor_sampler_retinenet = AnchorSamplerRetinaNet(cfgs)
self.refine_anchor_sampler_r3det_dcl = RefineAnchorSamplerR3DetDCL(
cfgs)
self.losses = LossDCL(self.cfgs)
self.coding_len = get_code_len(int(cfgs.ANGLE_RANGE / cfgs.OMEGA),
mode=cfgs.ANGLE_MODE)
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)