Results from a trained model of mine. about pytorch-unet HOT 8 CLOSED

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 :

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

from pytorch-unet.

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.

from pytorch-unet.

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!

from pytorch-unet.

@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...

from pytorch-unet.

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

from pytorch-unet.

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.

from pytorch-unet.

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

from pytorch-unet.

I have not recognized the repo was updated.

from pytorch-unet.