zjt/SAMapi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix incorrect shape in ResizeLongestSide.apply_image_torch.

Browse Source
This commit is contained in:
Eric Mintun 2023-04-15 10:40:48 -07:00
parent 02acff7859
commit 567662b0fd
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
segment_anything/utils/transforms.py
View File

@ -59,7 +59,7 @@ class ResizeLongestSide:
the transformation expected by the model. the transformation expected by the model.
""" """
# Expects an image in BCHW format. May not exactly match apply_image. # Expects an image in BCHW format. May not exactly match apply_image.
target_size = self.get_preprocess_shape(image.shape[0], image.shape[1], self.target_length) target_size = self.get_preprocess_shape(image.shape[2], image.shape[3], self.target_length)
return F.interpolate( return F.interpolate(
image, target_size, mode="bilinear", align_corners=False, antialias=True image, target_size, mode="bilinear", align_corners=False, antialias=True
) )