med-air / EndoNeRF

Neural Rendering for Stereo 3D Reconstruction of Deformable Tissues in Robotic Surgery
https://med-air.github.io/EndoNeRF/
196 stars 18 forks source link

I set use_fgmask = False,it get local variable 'ray_importance_maps' referenced before assignment #14

Closed sincerely1 closed 1 year ago

sincerely1 commented 1 year ago

hello,I want to train a NeRF without Mask, than I set use_fgmask=False,but it get " local variable 'ray_importance_maps' referenced before assignment" in run_endonerf.py at 1183 line. In the code, I found that load_llff_data function return mask none, so the ray_importance_maps not be assigned. I don't know how to deal with it.

yuehaowang commented 1 year ago

Hi, thanks for your question.

Our provided code didn't implement for use_fgmask=False yet. Nevertheless, you can disable tool mask-guided ray casting by creating dummy masks, i.e., full-black images.

sincerely1 commented 1 year ago

ok,thank for you help.