Neural Scene Flow Fields using pytorch-lightning with potential improvements
nsff_pl
Neural Scene Flow Fields using pytorch-lightning. This repo reimplements the NSFF idea, but modifies several operations based on observation of NSFF results and discussions with the authors. For discussion details, please see the issues of the original repo. The code is based on my previous implementation.
The main modifications are the followings:
- Remove the blending weight in static NeRF. I adopt the addition strategy in NeRF-W.
- Compose static dynamic also in image warping.
Implementation details are in models/rendering.py.
These modifications empirically produces better result on the kid-running scene, as shown below:
IMPORTANT: The code for kid-running scene is moved to nsff_orig branch (the images are still shown here just to showcase)! The master branch will be updated for custom data usage.