From 58552854938da0e6e755e6f3959debf16a88a3cf Mon Sep 17 00:00:00 2001
From: FeiGeChuanShu <774074168@qq.com>
Date: Tue, 26 Oct 2021 12:35:35 +0800
Subject: [PATCH 1/2] Update run_test.py

---
 run_test.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/run_test.py b/run_test.py
index 12c8e36..ac5d463 100644
--- a/run_test.py
+++ b/run_test.py
@@ -85,8 +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
@@ -99,4 +97,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)

From c33084c8b28932ce4f7c8c7efb2836bdacee881d Mon Sep 17 00:00:00 2001
From: FeiGeChuanShu <774074168@qq.com>
Date: Tue, 26 Oct 2021 13:10:40 +0800
Subject: [PATCH 2/2] Update run_test.py

---
 run_test.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/run_test.py b/run_test.py
index ac5d463..b8fe53c 100644
--- a/run_test.py
+++ b/run_test.py
@@ -85,7 +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_points = outputs['pred_points'][0]
     # draw the predictions
     size = 2
     img_to_draw = cv2.cvtColor(np.array(img_raw), cv2.COLOR_RGB2BGR)