def merge_outputs(self, detections): results = np.concatenate([detection[1] for detection in detections], axis=0).astype(np.float32) if self.cfg.TEST.NMS or len(self.cfg.TEST.TEST_SCALES) > 1: soft_nms_39(results, Nt=0.5, method=2) results = results.tolist() return results
def merge_outputs(self, detections): results = {} results[1] = np.concatenate([detection[1] for detection in detections], axis=0).astype(np.float32) if self.opt.nms or len(self.opt.test_scales) > 1: soft_nms_39(results[1], Nt=0.5, method=2) results[1] = results[1].tolist() return results