Project-MONAI / MONAI

AI Toolkit for Healthcare Imaging
https://monai.io/
Apache License 2.0
5.5k stars 1.01k forks source link

Fix the computation of the crop region end index #7825

Open AndersMollgaard opened 3 weeks ago

AndersMollgaard commented 3 weeks ago

Fixes #7824.

Description

The main impact of the PR is to change the computation of the end index for the extended box that RandCropBoxByPosNegLabeld uses to sample foreground centers in when self.whole_box = True. The problem and the fix is described in detail in the issue.

The PR furthermore makes a minor change to the computation of the start index of the extended box to ensure a center chosen at the margin will fully contain the original box.

I have not run the full test suite, but I have checked that I do not break the relevant unit-test: test_rand_crop_by_pos_neg_labeld.py

Types of changes