浏览代码

fix comments in convert_boxes_list_to_batched_boxes

quarrying 4 年之前
父节点
当前提交
5762021724
共有 1 个文件被更改,包括 1 次插入1 次删除
  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