search

NVlabs / stylegan2-ada-pytorch

StyleGAN2-ADA - Official PyTorch implementation
https://arxiv.org/abs/2006.06676
Other
4.07k stars 1.16k forks source link

Training for RGBD images? #113

Open nguyenmichau opened 3 years ago

nguyenmichau commented 3 years ago

Hello, does anyone know if it's possible to train with RGBD images using styleGAN2? Is there a way to modify the network so that it can work with an extra layer for the image? Thank you in advance for your help

dinoanasta commented 3 years ago

I had a similar issue since my dataset was RGBA images, but I ended up using ImageMagick to remove the alpha channel and create RGB images.

To install ImageMagick: sudo apt-get install imagemagick

The command I used was as follows: mogrify -flatten +matte -path ../EditedRGBImages/ ../OriginalRGBAImages/*.png

I know this doesn't answer your question directly, but it was a perfect workaround for me. I hope it helps.