Unified output names (WongKinYiu#283)

ShanksYI · Jul 29, 2022 · 6baceff · 6baceff
1 parent 5f7d38b
commit 6baceff
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/utils/add_nms.py b/utils/add_nms.py
@@ -111,22 +111,22 @@ def register_nms(
 
         # 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],
         )