From ef7039ea1806669c4cd4e8eb5abd3182a66883f9 Mon Sep 17 00:00:00 2001 From: Jonas Heinle Date: Thu, 23 Mar 2023 11:28:31 +0100 Subject: [PATCH] Error in tensor shape of docstring (#11206) Corrected the tensor shape in the doc string. The incoming masks are stacked in dim=0 therefore the doc is wrong Signed-off-by: Jonas Heinle --- utils/segment/general.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/segment/general.py b/utils/segment/general.py index 9da8945..f1b2f1d 100644 --- a/utils/segment/general.py +++ b/utils/segment/general.py @@ -10,7 +10,7 @@ def crop_mask(masks, boxes): Vectorized by Chong (thanks Chong). Args: - - masks should be a size [h, w, n] tensor of masks + - masks should be a size [n, h, w] tensor of masks - boxes should be a size [n, 4] tensor of bbox coords in relative point form """