Results from a trained model of mine.

[changjo]

Hi,
Thanks for your codes.
I trained the model with only changing batch size to 4 and I predicted some test images.
The right part in the image from a trained model is distorted and right parts in other results are also similar.
I was wondering why this happened and what your pre-trained parameters or any pre-processing except splitting a image to left and right are. Thanks.

• "Original image"
  

• Result from pre-trained model "MODEL.pth"
  

• Result from my trained model
  

[milesial]

Hello, that's strange. I didn't preprocess except for the rescaling and splitting.
I used the train script as it is, but with a batch size of 1 because of memory issues.

I don't remember the learning rate I used or the number of epochs, but you should try letting it run longer, with a different learning rate.

Also when predicting you can try to set the mask threshold to a higher value :

Pytorch-UNet/predict.py

Line 94 in 23e8173

parser.add_argument('--mask-threshold', '-t', type=float,

like 0.6 or more, so less pixels are considered white.

Last thing, have you tried using the CRF postprocessing ? That might help a bit.

[changjo]

Hi,
I used the CRF postprocessing but seems not to help. I also tried to change the mask threshold but not working..
I may try to change the learning rate and the number of epochs. Thanks.

[bermudezarii]

At the end, what did you do @changjo? Did you get a better result? Thanks!
Because I have the same problem as yours, also, where does the model from the new training is saved? Is the last checkpoint?

Thanks!

[changjo]

@bermudezarii I tried to change the learning rate, the training epochs, and batch size but I did not work... Currently, I have not been looking into the codes...

[LinglanZhao]

How can I get the pre-trained model "MODEL.pth"? (or I have to train a segmentation model from scratch?) @changjo @milesial

[milesial]

The pretrained model was available for the commit https://github.com/milesial/Pytorch-UNet/tree/e2e46ce509382a45b1db4e1f639aeed568f6cb3e and before. After that, I cleaned up the code and the model and I don't think the same file will load.

[LinglanZhao]

Thanks and MODEL.pth can be loaded successfully using the corresponding commit!

[changjo]

I have not recognized the repo was updated.