trained model (train_unet_demo) gives bad images.h5

[marinapapageorgiou]

Hello @mmuckley, I am studing singlecoil (Unet model) and knee.

I run the pretrained model (run_pretrained_unet_inference.py) and the results are these images:

After I trained the Unet model (train_unet_demo.py) and I tested it. The images of the testing are these:

If it is usuful I run the model with max_epochs=50 and lr=0.001, # learning rate. And i tried also with max_epochs=50 and lr=0.0003, # learning rate. But in both cases the best epoch that is keeps as the best is epoch=41. Also, I am using only these files for dataset knee_singlecoil_train knee_singlecoil_val knee_singlecoil_test

So my question is, What is wrong with the testing images of my training model? Is the dataset not big enough? Is the learning rate not appropriate? Is something else? Does anyone know?