-
Notifications
You must be signed in to change notification settings - Fork 1.1k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Added (Rand)ScaleScaleIntensityFixedMean(d) and modified (Rand)AdjustContrast(d) #6542
Conversation
…Contrast(d) with by adding arguments
for more information, see https://pre-commit.ci
Signed-off-by: Aaron Kujawa <askujawa@gmail.com>
Signed-off-by: Aaron Kujawa <askujawa@gmail.com>
/build |
Hi @KumoLiu , Could you please help review this PR? Thanks in advance. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for the PR, few comments inline.
Signed-off-by: Aaron Kujawa <askujawa@gmail.com>
Signed-off-by: Aaron Kujawa <askujawa@gmail.com>
Signed-off-by: Aaron Kujawa <askujawa@gmail.com>
Signed-off-by: Aaron Kujawa <askujawa@gmail.com>
Signed-off-by: Aaron Kujawa <askujawa@gmail.com>
Signed-off-by: Aaron Kujawa <askujawa@gmail.com>
Signed-off-by: Aaron Kujawa <askujawa@gmail.com>
Signed-off-by: Aaron Kujawa <askujawa@gmail.com>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Many thanks @aaronkujawa, I'm merging this after tagging v1.2.
/build |
Description
This PR adds the intensity transform ScaleIntensityFixedMean (including its random and random dictionary version) and modifies the intensity transform AdjustContrast and its dictionary version. It adds functionality available in the corresponding nnU-Net transforms ContrastAugmentationTransform and GammaTransform.
Specifically, ScaleIntensityFixedMean scales the intensity of the input image by a factor v = v * (1 + factor) (same as the existing ScaleIntensity transform when used with the factor argument). The added functionality is provided by two arguments:
fixed_mean: subtract the mean intensity before scaling with factor, then add the same value after scaling to ensure that the output has the same mean intensity as the input.
preserve_range: clips the output array/tensor to the range of the input array/tensor
AdjustContrast was modified by adding two arguments:
invert_image: multiplies all intensity values by -1 before gamma transform and again after gamma transform
retain_stats: applies a scaling factor and an offset to all intensity values after the gamma transform to ensure that the output intensity distribution has the same mean and standard deviation as the intensity distribution of the input
Types of changes
./runtests.sh -f -u --net --coverage
../runtests.sh --quick --unittests --disttests
.make html
command in thedocs/
folder.