Correct docstring and delete unused code in `scale_image` function (#4001)

single_channel
Flippchen 1 year ago committed by GitHub
parent c20d2654e9
commit 02a3cdd0e7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -321,7 +321,7 @@ def scale_image(masks, im0_shape, ratio_pad=None):
Takes a mask, and resizes it to the original image size Takes a mask, and resizes it to the original image size
Args: Args:
masks (torch.Tensor): resized and padded masks/images, [h, w, num]/[h, w, 3]. masks (np.ndarray): resized and padded masks/images, [h, w, num]/[h, w, 3].
im0_shape (tuple): the original image shape im0_shape (tuple): the original image shape
ratio_pad (tuple): the ratio of the padding to the original image. ratio_pad (tuple): the ratio of the padding to the original image.
@ -344,9 +344,6 @@ def scale_image(masks, im0_shape, ratio_pad=None):
if len(masks.shape) < 2: if len(masks.shape) < 2:
raise ValueError(f'"len of masks shape" should be 2 or 3, but got {len(masks.shape)}') raise ValueError(f'"len of masks shape" should be 2 or 3, but got {len(masks.shape)}')
masks = masks[top:bottom, left:right] masks = masks[top:bottom, left:right]
# masks = masks.permute(2, 0, 1).contiguous()
# masks = F.interpolate(masks[None], im0_shape[:2], mode='bilinear', align_corners=False)[0]
# masks = masks.permute(1, 2, 0).contiguous()
masks = cv2.resize(masks, (im0_shape[1], im0_shape[0])) masks = cv2.resize(masks, (im0_shape[1], im0_shape[0]))
if len(masks.shape) == 2: if len(masks.shape) == 2:
masks = masks[:, :, None] masks = masks[:, :, None]

Loading…
Cancel
Save