search

wbhu / Tri-MipRF

Tri-MipRF: Tri-Mip Representation for Efficient Anti-Aliasing Neural Radiance Fields, ICCV'23 (Oral, Best Paper Finalist)
https://wbhu.github.io/projects/Tri-MipRF
451 stars 13 forks source link

cuda error! #28

Closed zf-666 closed 1 year ago

zf-666 commented 1 year ago 
During handling of the above exception, another exception occurred:

  File "/home/zhoufan/Tri-MipRF/neural_field/field/trimipRF.py", line 97, in query_rgb
    self.mlp_head(h)
  File "/home/zhoufan/Tri-MipRF/neural_field/model/trimipRF.py", line 115, in rgb_sigma_fn
    rgb = self.field.query_rgb(dir=t_dirs, embedding=feature)['rgb']
  File "/home/zhoufan/Tri-MipRF/neural_field/model/trimipRF.py", line 140, in rendering
    rgbs, sigmas = rgb_sigma_fn(t_starts, t_ends, ray_indices.long())
  File "/home/zhoufan/Tri-MipRF/neural_field/model/trimipRF.py", line 118, in forward
    return self.rendering(
  File "/home/zhoufan/Tri-MipRF/trainer/trainer.py", line 169, in eval_img
    rb = self.model(
  File "/home/zhoufan/Tri-MipRF/trainer/trainer.py", line 141, in fit
    metrics, final_rb, target = self.eval_img(
  File "/home/zhoufan/Tri-MipRF/main.py", line 68, in main
    trainer.fit()
  File "/home/zhoufan/Tri-MipRF/main.py", line 108, in <module>
    main()

eval的时候，trimipRF.py 里面的 = nerfacc.ray_marching()返回的是空值，而训练的时候是正常的。