NVlabs / imaginaire

NVIDIA's Deep Imagination Team's PyTorch Library
Other
3.99k stars 444 forks source link

Faster, Smaller and Efficient Pix2pix #183

Open spikejones1040 opened 10 months ago

spikejones1040 commented 10 months ago

Hi,

I have been using pix2pix model for image-to-image translation. I am happy with the quality but unfortunately it consumes lot of memory (model size is about 2.8 GB). I am interested in generators that are faster, memory efficient but can retain visual quality. I know pix2pix was released 4-5 years ago and was wondering if there were any models that fit this criteria?

I checked models from model-zoo from imaginaire, but all of them are large in terms of size and memory. Any help will be greatly appreciated.

Thanks