From 6bacefff5ce5f0ea3ea7c5ce7e003347919c11a7 Mon Sep 17 00:00:00 2001
From: tripleMu <gpu@163.com>
Date: Fri, 29 Jul 2022 09:58:57 +0800
Subject: [PATCH] Unified output names (#283)

---
 utils/add_nms.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/utils/add_nms.py b/utils/add_nms.py
index 927355e..0a1f797 100644
--- a/utils/add_nms.py
+++ b/utils/add_nms.py
@@ -111,22 +111,22 @@ class RegisterNMS(object):
 
         # NMS Outputs
         output_num_detections = gs.Variable(
-            name="num_detections",
+            name="num_dets",
             dtype=np.int32,
             shape=[self.batch_size, 1],
         )  # A scalar indicating the number of valid detections per batch image.
         output_boxes = gs.Variable(
-            name="detection_boxes",
+            name="det_boxes",
             dtype=dtype_output,
             shape=[self.batch_size, detections_per_img, 4],
         )
         output_scores = gs.Variable(
-            name="detection_scores",
+            name="det_scores",
             dtype=dtype_output,
             shape=[self.batch_size, detections_per_img],
         )
         output_labels = gs.Variable(
-            name="detection_classes",
+            name="det_classes",
             dtype=np.int32,
             shape=[self.batch_size, detections_per_img],
         )