예제 #1
0
 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)
예제 #2
0
 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)