Fictionarry / ER-NeRF

[ICCV'23] Efficient Region-Aware Neural Radiance Fields for High-Fidelity Talking Portrait Synthesis
https://fictionarry.github.io/ER-NeRF/
MIT License
893 stars 124 forks source link

'eye_att' referenced before assignment #145

Open lcyuanjiang opened 1 month ago

G-force78 commented 1 month ago

https://github.com/Fictionarry/ER-NeRF/blob/c8ce2ec1689bb6050f5567c5dd66320d43cbe2c1/nerf_triplane/network.py#L288

Add eye_att = torch.tensor([])

cacard commented 2 weeks ago

https://github.com/Fictionarry/ER-NeRF/blob/c8ce2ec1689bb6050f5567c5dd66320d43cbe2c1/nerf_triplane/network.py#L288

Add eye_att = torch.tensor([])

will new error:

File "/root/_ai/ER-NeRF/nerf_triplane/utils.py", line 730, in train_step eye_mask = data['eye_mask'] # [B, N] KeyError: 'eye_mask'