batchgenerators icon indicating copy to clipboard operation
batchgenerators copied to clipboard

10x performance improvement of DownsampleSegForDSTransform2 in nnunet by moving 1 line of code

Open AwesomeLemon opened this issue 2 years ago • 0 comments

Hi,

Finding unique labels is not necessary when order == 0. In DownsampleSegForDSTransform2 in nnunet, this function is called multiple times, and it all adds up, especially in the limited CPU scenario with high-res images. By introducing this change, I've reduced the time of this transform from ~2 seconds to ~0.2 seconds.

Best,

AwesomeLemon avatar Jan 29 '24 12:01 AwesomeLemon