Pretrained model

[QIQIVTrackingY]

Can you provide me a pretrained model? Thank you very much

[milesial]

If enough people want this, I could run a training on the Carvana dataset and share the weights. But anyone with a NVIDIA GPU could train the model on it in a few hours.

EDIT: see below for the pretrained model

[VictorZoo]

Do we have any pre-trained models?

[milesial]

@VictorZoo Not yet, do you want one for the Carvana dataset?

[Nikronic]

Hi, as I used this repo a long time ago, I could remember there was a MODEL.pth file. So, I searched for it and here is the link:
https://github.com/milesial/Pytorch-UNet/blob/e2e46ce509382a45b1db4e1f639aeed568f6cb3e/MODEL.pth

[milesial]

Warning, the repo at that version was quite buggy, and this model is not compatible with the current version.

[milesial]

I will train a Carvana model at 10 reactions on the original comment.

[VictorZoo]

I will train a Carvana model at 10 reactions on the original comment.

Okay! Thank you very much!

[buptlxy]

I want a pretrained model too! Thank you very much!

[VladimirBychkov]

Please add pretrained model! Thanks!

[msnh2012]

+1

[fitsir]

I want a pretrained model too! Thank you very much!

[ninanikitina]

Hi! I want the model too! Thank you!

[milesial]

Hello everyone, the Carvana model is available in the releases.
This was trained for 5 epochs, with scale=1 and bilinear=True. Let me know if you want one with transposed convs.

You can also use torch.hub to load it:

torch.hub.load('milesial/Pytorch-UNet', 'unet_carvana')

More info about torch.hub