Просмотр исходного кода

fix comments in convert_boxes_list_to_batched_boxes

quarrying 4 лет назад
Родитель
Сommit
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