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
