MIC-DKFZ / medicaldetectiontoolkit

The Medical Detection Toolkit contains 2D + 3D implementations of prevalent object detectors such as Mask R-CNN, Retina Net, Retina U-Net, as well as a training and inference framework focused on dealing with medical images.
Apache License 2.0
1.31k stars 297 forks source link

pyramid_roi_align in MaskRCNN if the roi_level of level P6 is included #84

Open 13917904019 opened 5 years ago

13917904019 commented 5 years ago

Hello

In your MaskRCNN work, if Pyramid contains additional level P6, the roi_level of level P6 should be 4,not 5. Am I right? But in mrcnn.py, in about 400 lines, 'roi_level[hw > 0.65] = 5' should be 'roi_level[hw > 0.65] = 4'. Is it right?

By the way, your work is great.