From 3db86c05f69d310a55166ff4cad2e4050eb01355 Mon Sep 17 00:00:00 2001 From: Hao Zhang <931851825@qq.com> Date: Fri, 7 Apr 2023 18:47:42 +0800 Subject: [PATCH] debug maskdino --- projects/maskdino/maskdino.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/projects/maskdino/maskdino.py b/projects/maskdino/maskdino.py index 1e174df4..75496f61 100644 --- a/projects/maskdino/maskdino.py +++ b/projects/maskdino/maskdino.py @@ -187,8 +187,8 @@ def forward(self, batched_inputs): for mask_cls_result, mask_pred_result, mask_box_result, input_per_image, image_size in zip( mask_cls_results, mask_pred_results, mask_box_results, batched_inputs, images.image_sizes ): # image_size is augmented size, not divisible to 32 - height = input_per_image.get("height", image_size[0]) # real size - width = input_per_image.get("width", image_size[1]) + height = input_per_image["height"]#, image_size[0]) # real size + width = input_per_image["width"]#, image_size[1]) processed_results.append({}) new_size = mask_pred_result.shape[-2:] # padded size (divisible to 32)