├── .gitignore ├── README.md ├── SR ├── models.py └── sr_model_zoo │ ├── edsr.py │ ├── eg3d.py │ ├── sr_backbones_utils.py │ ├── stylegan_sr.py │ └── unet.py ├── convert.py ├── convert.sh ├── create_traj ├── bezier_curve.py ├── create_interp_traj.py └── random_poses.py ├── datasets ├── __init__.py ├── base.py ├── colmap.py ├── colmap_utils.py ├── color_utils.py ├── depth_utils.py ├── nerf.py ├── nerfpp.py ├── nsvf.py ├── ray_utils.py └── rtmv.py ├── losses.py ├── metrics.py ├── models ├── __init__.py ├── csrc │ ├── binding.cpp │ ├── include │ │ ├── helper_math.h │ │ └── utils.h │ ├── intersection.cu │ ├── losses.cu │ ├── raymarching.cu │ ├── setup.py │ ├── volumerendering.cu │ └── volumerendering_feat.cu ├── custom_functions.py ├── networks.py ├── project.py └── rendering.py ├── opt.py ├── render_traj.sh ├── requirements.txt ├── scripts ├── Synthetic_NeRF │ ├── final_eval │ │ └── eval_Chair.sh │ ├── hparam_tuning │ │ ├── lr_schedule_tuning.sh │ │ └── lr_tuning.sh │ ├── joint_training │ │ ├── direct_E2E_Chair.sh │ │ └── eval_Chair.sh │ └── pretraining │ │ ├── eval_Chair.sh │ │ ├── pretrain_Chair.sh │ │ └── pretrain_Hotdog.sh └── TanksAndTemple │ ├── joint_training │ ├── direct_E2E_Barn.sh │ ├── direct_E2E_Cater.sh │ ├── direct_E2E_Family.sh │ ├── direct_E2E_Igna.sh │ └── direct_E2E_Truck.sh │ └── pretraining │ ├── pretrain_Barn.sh │ ├── pretrain_Cater.sh │ ├── pretrain_Family.sh │ ├── pretrain_Igna.sh │ └── pretrain_Truck.sh ├── train.py ├── trt.sh └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/README.md -------------------------------------------------------------------------------- /SR/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/SR/models.py -------------------------------------------------------------------------------- /SR/sr_model_zoo/edsr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/SR/sr_model_zoo/edsr.py -------------------------------------------------------------------------------- /SR/sr_model_zoo/eg3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/SR/sr_model_zoo/eg3d.py -------------------------------------------------------------------------------- /SR/sr_model_zoo/sr_backbones_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/SR/sr_model_zoo/sr_backbones_utils.py -------------------------------------------------------------------------------- /SR/sr_model_zoo/stylegan_sr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/SR/sr_model_zoo/stylegan_sr.py -------------------------------------------------------------------------------- /SR/sr_model_zoo/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/SR/sr_model_zoo/unet.py -------------------------------------------------------------------------------- /convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/convert.py -------------------------------------------------------------------------------- /convert.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/convert.sh -------------------------------------------------------------------------------- /create_traj/bezier_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/create_traj/bezier_curve.py -------------------------------------------------------------------------------- /create_traj/create_interp_traj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/create_traj/create_interp_traj.py -------------------------------------------------------------------------------- /create_traj/random_poses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/create_traj/random_poses.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/datasets/base.py -------------------------------------------------------------------------------- /datasets/colmap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/datasets/colmap.py -------------------------------------------------------------------------------- /datasets/colmap_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/datasets/colmap_utils.py -------------------------------------------------------------------------------- /datasets/color_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/datasets/color_utils.py -------------------------------------------------------------------------------- /datasets/depth_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/datasets/depth_utils.py -------------------------------------------------------------------------------- /datasets/nerf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/datasets/nerf.py -------------------------------------------------------------------------------- /datasets/nerfpp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/datasets/nerfpp.py -------------------------------------------------------------------------------- /datasets/nsvf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/datasets/nsvf.py -------------------------------------------------------------------------------- /datasets/ray_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/datasets/ray_utils.py -------------------------------------------------------------------------------- /datasets/rtmv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/datasets/rtmv.py -------------------------------------------------------------------------------- /losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/losses.py -------------------------------------------------------------------------------- /metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/metrics.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/csrc/binding.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/models/csrc/binding.cpp -------------------------------------------------------------------------------- /models/csrc/include/helper_math.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/models/csrc/include/helper_math.h -------------------------------------------------------------------------------- /models/csrc/include/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/models/csrc/include/utils.h -------------------------------------------------------------------------------- /models/csrc/intersection.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/models/csrc/intersection.cu -------------------------------------------------------------------------------- /models/csrc/losses.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/models/csrc/losses.cu -------------------------------------------------------------------------------- /models/csrc/raymarching.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/models/csrc/raymarching.cu -------------------------------------------------------------------------------- /models/csrc/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/models/csrc/setup.py -------------------------------------------------------------------------------- /models/csrc/volumerendering.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/models/csrc/volumerendering.cu -------------------------------------------------------------------------------- /models/csrc/volumerendering_feat.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/models/csrc/volumerendering_feat.cu -------------------------------------------------------------------------------- /models/custom_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/models/custom_functions.py -------------------------------------------------------------------------------- /models/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/models/networks.py -------------------------------------------------------------------------------- /models/project.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/models/project.py -------------------------------------------------------------------------------- /models/rendering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/models/rendering.py -------------------------------------------------------------------------------- /opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/opt.py -------------------------------------------------------------------------------- /render_traj.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/render_traj.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/Synthetic_NeRF/final_eval/eval_Chair.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/scripts/Synthetic_NeRF/final_eval/eval_Chair.sh -------------------------------------------------------------------------------- /scripts/Synthetic_NeRF/hparam_tuning/lr_schedule_tuning.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/scripts/Synthetic_NeRF/hparam_tuning/lr_schedule_tuning.sh -------------------------------------------------------------------------------- /scripts/Synthetic_NeRF/hparam_tuning/lr_tuning.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/scripts/Synthetic_NeRF/hparam_tuning/lr_tuning.sh -------------------------------------------------------------------------------- /scripts/Synthetic_NeRF/joint_training/direct_E2E_Chair.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/scripts/Synthetic_NeRF/joint_training/direct_E2E_Chair.sh -------------------------------------------------------------------------------- /scripts/Synthetic_NeRF/joint_training/eval_Chair.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/scripts/Synthetic_NeRF/joint_training/eval_Chair.sh -------------------------------------------------------------------------------- /scripts/Synthetic_NeRF/pretraining/eval_Chair.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/scripts/Synthetic_NeRF/pretraining/eval_Chair.sh -------------------------------------------------------------------------------- /scripts/Synthetic_NeRF/pretraining/pretrain_Chair.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/scripts/Synthetic_NeRF/pretraining/pretrain_Chair.sh -------------------------------------------------------------------------------- /scripts/Synthetic_NeRF/pretraining/pretrain_Hotdog.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/scripts/Synthetic_NeRF/pretraining/pretrain_Hotdog.sh -------------------------------------------------------------------------------- /scripts/TanksAndTemple/joint_training/direct_E2E_Barn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/scripts/TanksAndTemple/joint_training/direct_E2E_Barn.sh -------------------------------------------------------------------------------- /scripts/TanksAndTemple/joint_training/direct_E2E_Cater.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/scripts/TanksAndTemple/joint_training/direct_E2E_Cater.sh -------------------------------------------------------------------------------- /scripts/TanksAndTemple/joint_training/direct_E2E_Family.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/scripts/TanksAndTemple/joint_training/direct_E2E_Family.sh -------------------------------------------------------------------------------- /scripts/TanksAndTemple/joint_training/direct_E2E_Igna.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/scripts/TanksAndTemple/joint_training/direct_E2E_Igna.sh -------------------------------------------------------------------------------- /scripts/TanksAndTemple/joint_training/direct_E2E_Truck.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/scripts/TanksAndTemple/joint_training/direct_E2E_Truck.sh -------------------------------------------------------------------------------- /scripts/TanksAndTemple/pretraining/pretrain_Barn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/scripts/TanksAndTemple/pretraining/pretrain_Barn.sh -------------------------------------------------------------------------------- /scripts/TanksAndTemple/pretraining/pretrain_Cater.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/scripts/TanksAndTemple/pretraining/pretrain_Cater.sh -------------------------------------------------------------------------------- /scripts/TanksAndTemple/pretraining/pretrain_Family.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/scripts/TanksAndTemple/pretraining/pretrain_Family.sh -------------------------------------------------------------------------------- /scripts/TanksAndTemple/pretraining/pretrain_Igna.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/scripts/TanksAndTemple/pretraining/pretrain_Igna.sh -------------------------------------------------------------------------------- /scripts/TanksAndTemple/pretraining/pretrain_Truck.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/scripts/TanksAndTemple/pretraining/pretrain_Truck.sh -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/train.py -------------------------------------------------------------------------------- /trt.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/trt.sh -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JasonLSC/SteerNeRF_official/HEAD/utils.py --------------------------------------------------------------------------------