Skip to content

GitLab

Unverified Commit f58f1d1c authored by Hanzi Mao's avatar Hanzi Mao Committed by GitHub
Browse files

Merge pull request #58 from EndingCredits/main 

Fix baking original input size values to ONNX graph
parents efeab729 e9f2d580
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
segment_anything/utils/onnx.py
View file @ f58f1d1c
...@@ -81,8 +81,8 @@ class SamOnnxModel(nn.Module): ...@@ -81,8 +81,8 @@ class SamOnnxModel(nn.Module):
align_corners=False, align_corners=False,
) )
prepadded_size = self.resize_longest_image_size(orig_im_size, self.img_size) prepadded_size = self.resize_longest_image_size(orig_im_size, self.img_size).to(torch.int64)
masks = masks[..., : int(prepadded_size[0]), : int(prepadded_size[1])] masks = masks[..., : prepadded_size[0], : prepadded_size[1]]
orig_im_size = orig_im_size.to(torch.int64) orig_im_size = orig_im_size.to(torch.int64)
h, w = orig_im_size[0], orig_im_size[1] h, w = orig_im_size[0], orig_im_size[1]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment