search

lucidrains / pi-GAN-pytorch

Implementation of π-GAN, for 3d-aware image synthesis, in Pytorch
MIT License
115 stars 14 forks source link

training script #3

Open kam1107 opened 3 years ago

kam1107 commented 3 years ago

Hi :) I am new to nerf and am having trouble identifying 'ray_direction'. Can you provide an example of this variable (or the shape of it)? It would be of great help if there is an example training script. Many thanks!

lucidrains commented 3 years ago

@kam1107 Hi! You can pass in the ray direction here https://github.com/lucidrains/pi-GAN-pytorch/blob/main/pi_gan_pytorch/pi_gan_pytorch.py#L158