search

IDEA-Research / detrex

detrex is a research platform for DETR-based object detection, segmentation, pose estimation and other visual recognition tasks.
https://detrex.readthedocs.io/en/latest/
Apache License 2.0
1.92k stars 202 forks source link

Different from the original code of MaskDINO #264

Open winpih opened 1 year ago

winpih commented 1 year ago

  1. maskdino_encoder.py of detrex line 303

    # stride = 4, self.common_stride = 4 -> self.num_fpn_levels = 0
self.num_fpn_levels = int(np.log2(stride) - np.log2(self.common_stride))

  2. maskdino_encoder.py of Original MaskDINO line 305

    # stride = 4, self.common_stride = 4 -> self.num_fpn_levels = 1
self.num_fpn_levels = max(int(np.log2(stride) - np.log2(self.common_stride)), 1)

The size of "pred_masks" varies depending on the difference in the above code. Please check it.

rentainhe commented 1 year ago

Thanks for reporting this issue, we will refine it carefully in the future release