diff --git a/run_test.py b/run_test.py index 12c8e36..b8fe53c 100644 --- a/run_test.py +++ b/run_test.py @@ -85,9 +85,6 @@ def main(args, debug=False): points = outputs_points[outputs_scores > threshold].detach().cpu().numpy().tolist() predict_cnt = int((outputs_scores > threshold).sum()) - outputs_scores = torch.nn.functional.softmax(outputs['pred_logits'], -1)[:, :, 1][0] - - outputs_points = outputs['pred_points'][0] # draw the predictions size = 2 img_to_draw = cv2.cvtColor(np.array(img_raw), cv2.COLOR_RGB2BGR) @@ -99,4 +96,4 @@ def main(args, debug=False): if __name__ == '__main__': parser = argparse.ArgumentParser('P2PNet evaluation script', parents=[get_args_parser()]) args = parser.parse_args() - main(args) \ No newline at end of file + main(args)