Malav Bateriwala
Malav Bateriwala
Which model is trans-Unet model? Can you explain more about the data and transformation you are using?
Using R2Unet can go wrong as recurrent blocks do not learn that well in some cases. Try running the nested Unet model and check the results.
Won't be addding new features in this code now. But if you add a new model send me pull request.
Can you share more details on the error? (Line and input data shape)
Can you check your mean/std values and your image channels during the transform function?
It just requires different folder for all types of images. (3 channel image) You can modify it to use a csv.