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