search

qubvel / segmentation_models

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.
MIT License
4.73k stars 1.03k forks source link

Transfer weights from PyTorch U-Net to Keras U-Net #489

Open patrickselle opened 3 years ago

patrickselle commented 3 years ago

Hi,

Have a U-Net (encoder: reset50) trained but with PyTorch (link) and extracted the weights of the encoder and decoder. I would like to load these into the same net (U-Net with a resnet50 encoder) but in Keras (link). What is the easiest way to do this?

Best regards patrick