Przeglądaj źródła

fix comments in convert_boxes_list_to_batched_boxes

quarrying 4 lat temu
rodzic
commit
5762021724
1 zmienionych plików z 1 dodań i 1 usunięć
  1. 1 1
      khandy/boxes/batched_boxes.py

+ 1 - 1
khandy/boxes/batched_boxes.py

@@ -7,7 +7,7 @@ def convert_boxes_list_to_batched_boxes(boxes_list):
         boxes_list: list or tuple of ndarray with shape (N_i, 4+K)
         
     Returns:
-        ndarray with shape (M, 5+K)
+        ndarray with shape (M, 5+K) where M is sum of N_i.
         
     References:
         `convert_boxes_to_roi_format` in TorchVision