xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 21:44:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Flippchen 2023-07-28 14:52:36 +02:00 committed by GitHub
parent c20d2654e9
commit 02a3cdd0e7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ultralytics/utils/ops.py
View File

@ -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]