From f4311eb39016a2a05728282e6e4f24404c5b52ba Mon Sep 17 00:00:00 2001 From: Yonghye Kwon Date: Sat, 22 Jun 2024 18:27:30 +0900 Subject: [PATCH] set device via image tensor --- depth_anything_v2/dpt.py | 1 + 1 file changed, 1 insertion(+) diff --git a/depth_anything_v2/dpt.py b/depth_anything_v2/dpt.py index 18d3e6f8..908666a2 100644 --- a/depth_anything_v2/dpt.py +++ b/depth_anything_v2/dpt.py @@ -186,6 +186,7 @@ def forward(self, x): @torch.no_grad() def infer_image(self, raw_image, input_size=518): image, (h, w) = self.image2tensor(raw_image, input_size) + self.to(image.device) depth = self.forward(image)