From 184495fcde7971539a1fb78f52247240bfb79c8e Mon Sep 17 00:00:00 2001 From: George Grigorev Date: Wed, 23 Aug 2023 11:48:31 +0100 Subject: [PATCH] update zoedepth to be compatible with torch 2.1 --- zoedepth/models/base_models/midas.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/zoedepth/models/base_models/midas.py b/zoedepth/models/base_models/midas.py index e26f85895..36f6c3794 100644 --- a/zoedepth/models/base_models/midas.py +++ b/zoedepth/models/base_models/midas.py @@ -170,7 +170,7 @@ def get_size(self, width, height): 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) + 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):