🐛 fix intensity augmentation training

dyollb / segmantic

Semantic segmentation and image-to-image translation based on AI

https://dyollb.github.io/segmantic/

MIT License

1 stars 3 forks source link

Closed dyollb closed 2 years ago

dyollb commented 2 years ago

fix intensity augmentation training

recent version of monai (0.9) seems to introduce something that breaks the pytorch lightning training because of incompatible tensor types (issue 4537)
fixed use of RandCropByLabelClassesd to not provide an image_key (we already crop foreground)