pytorch-nested-unet
pytorch-nested-unet copied to clipboard
this repo is unable to run!
I have tried every step exactly as the author recorded. However, there is always a bug in the period of running train.py. I suggest all of the users to move to other repos!!!!!!!!!!!!
if you have this problem named "module albumentations.augmentations.transforms has no attribute RandomRotate90 error" solution :
- add this in the top of train.py-->>“import albumentations as A” 2 change “transform.RandomRotate90()” to “A.RandomRotate90()”
- and rewrite all “transform.Resize” to “A.Resize” in train.py
I tried using torch's transforms to replace this module, and it also did not work properly. I eventually found another blog with tensorflow framework, and adjusted it to both UNet and UNet++ models. Thanks for your reply.