Autoencoder with Transfer Learning?

Question

Is there a way I can train an autoencoder model using a pre-trained model like ResNet?

I'm trying to train an autoencoder model with input as an image and output as a masked version of that image.

Is it possible to use weights from a pretrained model here?

Answer 1

Yes! you can definitely do transfer learning using a pre-trained network, i.e. ResNet50 as the encoder in an autoencoder. For reference, check out the following link. https://github.com/hsinyilin19/ResNetVAE