autonomousvision / giraffe

This repository contains the code for the CVPR 2021 paper "GIRAFFE: Representing Scenes as Compositional Generative Neural Feature Fields"
https://m-niemeyer.github.io/project-pages/giraffe/index.html
MIT License
1.23k stars 160 forks source link

About fitting the latent codes of a given image #41

Closed Pbb-land closed 2 years ago

Pbb-land commented 2 years ago

Hello, thanks for your great work. I tried to optimize the latent codes of a given face image with your pre-trained CelebAHQ model but failed with 'CUDA out of memory'(on a single TITAN xp GPU,12G). I simply use the rgb l1 loss between the gt image and generate image, Adam optimizer. Could you please offer some official code or some strategy about solving this problem?

m-niemeyer commented 2 years ago

Hi, our repo does not support latent code optimization; I think the direction you took sounds reasonable as a start.