Added cast for midas.py call which was failing to recognize that int64 was a int.

2024-01-25 01:22:26 -07:00
parent 23d46bdecb
commit 27332ed5f0
1 changed files with 2 additions and 1 deletions
@@ -170,7 +170,8 @@ class Resize(object):

    def __call__(self, x):
        width, height = self.get_size(*x.shape[-2:][::-1])
-        return nn.functional.interpolate(x, (height, width), mode='bilinear', align_corners=True)
+        # Casting Correction.  int64 != int
+        return nn.functional.interpolate(x, (int(height), int(width)), mode='bilinear', align_corners=True)

 class PrepForMidas(object):
    def __init__(self, resize_mode="minimal", keep_aspect_ratio=True, img_size=384, do_resize=True):