Fix lint.

2023-04-10 09:21:25 -07:00
parent ca981bf377
commit 06bd20da89
2 changed files with 5 additions and 5 deletions
--- a/segment_anything/utils/onnx.py
+++ b/segment_anything/utils/onnx.py
@@ -82,7 +82,7 @@ class SamOnnxModel(nn.Module):
        )

        prepadded_size = self.resize_longest_image_size(orig_im_size, self.img_size).to(torch.int64)
-        masks = masks[..., : prepadded_size[0], : prepadded_size[1]]
+        masks = masks[..., : prepadded_size[0], : prepadded_size[1]]  # type: ignore

        orig_im_size = orig_im_size.to(torch.int64)
        h, w = orig_im_size[0], orig_im_size[1]