From 3aaa921ddd0f8ef39217201e4dd5ce5d78ca2ea5 Mon Sep 17 00:00:00 2001 From: Fabian Schuetze Date: Wed, 5 Apr 2023 12:43:33 +0200 Subject: [PATCH] remove width and height recalculation --- projects/maskdino/maskdino.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/projects/maskdino/maskdino.py b/projects/maskdino/maskdino.py index 1e174df4..974b020e 100644 --- a/projects/maskdino/maskdino.py +++ b/projects/maskdino/maskdino.py @@ -217,8 +217,6 @@ def forward(self, batched_inputs): # import ipdb; ipdb.set_trace() if self.instance_on: mask_box_result = mask_box_result.to(mask_pred_result) - height = new_size[0]/image_size[0]*height - width = new_size[1]/image_size[1]*width mask_box_result = self.box_postprocess(mask_box_result, height, width) instance_r = retry_if_cuda_oom(self.instance_inference)(mask_cls_result, mask_pred_result, mask_box_result)