pytorch-nested-unet icon indicating copy to clipboard operation
pytorch-nested-unet copied to clipboard

this repo is unable to run!

Open MorleyOlsen opened this issue 1 year ago • 2 comments

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!!!!!!!!!!!!

MorleyOlsen avatar Mar 22 '24 08:03 MorleyOlsen

if you have this problem named "module albumentations.augmentations.transforms has no attribute RandomRotate90 error" solution :

  1. add this in the top of train.py-->>“import albumentations as A” 2 change “transform.RandomRotate90()” to “A.RandomRotate90()”
  2. and rewrite all “transform.Resize” to “A.Resize” in train.py

pantongwang avatar Mar 25 '24 08:03 pantongwang

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.

MorleyOlsen avatar Mar 25 '24 08:03 MorleyOlsen