├── .gitignore ├── LICENSE.md ├── data ├── __init__.py ├── base_dataset.py └── dtu_dataset.py ├── models ├── __init__.py ├── atlasnet │ ├── __init__.py │ ├── inverse.py │ ├── networks.py │ └── ops │ │ ├── __init__.py │ │ ├── ball_query.py │ │ ├── chamfer.py │ │ ├── cuda │ │ ├── ball_query.cpp │ │ ├── ball_query_kernel.cu │ │ ├── chamfer.cpp │ │ ├── chamfer_kernel.cu │ │ ├── emd.cpp │ │ ├── emd_kernel.cu │ │ ├── fps.cpp │ │ ├── fps_kernel.cu │ │ ├── group_points.cpp │ │ ├── group_points_kernel.cu │ │ ├── knn_distance.cpp │ │ ├── knn_distance_kernel.cu │ │ ├── nn_distance.cpp │ │ └── nn_distance_kernel.cu │ │ ├── emd.py │ │ ├── fps.py │ │ ├── group_points.py │ │ ├── knn_distance.py │ │ ├── nn_distance.py │ │ ├── setup.py │ │ └── tests │ │ ├── test_ball_query.py │ │ ├── test_chamfer.py │ │ ├── test_fps.py │ │ ├── test_knn_distance.py │ │ └── test_nn_distance.py ├── base_decoder.py ├── base_model.py ├── decoder │ ├── __init__.py │ ├── atlas_cycle_decoder.py │ ├── conv_template.py │ ├── cycle_decoder.py │ ├── geometry_mlp_decoder.py │ ├── geometry_volume_decoder.py │ ├── original_nerf.py │ └── standard_volume_decoder.py ├── diff_ray_marching.py ├── diff_render_func.py ├── diff_transforms.py ├── embedding │ ├── __init__.py │ ├── gaussian_embedding.py │ └── lp_embedding.py ├── encoders.py ├── nerf_atlas_radiance_model.py ├── nerf_decoders.py ├── networks.py ├── point_decoder.py ├── texture │ ├── __init__.py │ └── texture_mlp.py ├── texture_networks.py └── volumetric_networks.py ├── options ├── __init__.py ├── base_options.py ├── test_options.py └── train_options.py ├── readme.md ├── run ├── DTU │ └── scan114 │ │ └── trainData │ │ ├── data.hdf5 │ │ ├── in_bboxes.npy │ │ ├── in_camAts.npy │ │ ├── in_camExtrinsics.npy │ │ ├── in_camFocal.npy │ │ ├── in_camOrgs.npy │ │ ├── in_camPrincpt.npy │ │ ├── in_camUps.npy │ │ ├── mesh_unit.ply │ │ └── pcd_down_unit.ply ├── dtu-freeze.sh ├── dtu.sh ├── dtu.vis.sh ├── train.py └── visualize_nerf_atlas_radiance.py └── utils ├── __init__.py ├── camera.py ├── cube_map.py ├── format.py ├── grid.py ├── grid_viewer.py ├── icosphere.py ├── string_util.py └── visualizer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/LICENSE.md -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/data/base_dataset.py -------------------------------------------------------------------------------- /data/dtu_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/data/dtu_dataset.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/atlasnet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/__init__.py -------------------------------------------------------------------------------- /models/atlasnet/inverse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/inverse.py -------------------------------------------------------------------------------- /models/atlasnet/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/networks.py -------------------------------------------------------------------------------- /models/atlasnet/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/atlasnet/ops/ball_query.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/ball_query.py -------------------------------------------------------------------------------- /models/atlasnet/ops/chamfer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/chamfer.py -------------------------------------------------------------------------------- /models/atlasnet/ops/cuda/ball_query.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/cuda/ball_query.cpp -------------------------------------------------------------------------------- /models/atlasnet/ops/cuda/ball_query_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/cuda/ball_query_kernel.cu -------------------------------------------------------------------------------- /models/atlasnet/ops/cuda/chamfer.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/cuda/chamfer.cpp -------------------------------------------------------------------------------- /models/atlasnet/ops/cuda/chamfer_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/cuda/chamfer_kernel.cu -------------------------------------------------------------------------------- /models/atlasnet/ops/cuda/emd.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/cuda/emd.cpp -------------------------------------------------------------------------------- /models/atlasnet/ops/cuda/emd_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/cuda/emd_kernel.cu -------------------------------------------------------------------------------- /models/atlasnet/ops/cuda/fps.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/cuda/fps.cpp -------------------------------------------------------------------------------- /models/atlasnet/ops/cuda/fps_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/cuda/fps_kernel.cu -------------------------------------------------------------------------------- /models/atlasnet/ops/cuda/group_points.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/cuda/group_points.cpp -------------------------------------------------------------------------------- /models/atlasnet/ops/cuda/group_points_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/cuda/group_points_kernel.cu -------------------------------------------------------------------------------- /models/atlasnet/ops/cuda/knn_distance.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/cuda/knn_distance.cpp -------------------------------------------------------------------------------- /models/atlasnet/ops/cuda/knn_distance_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/cuda/knn_distance_kernel.cu -------------------------------------------------------------------------------- /models/atlasnet/ops/cuda/nn_distance.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/cuda/nn_distance.cpp -------------------------------------------------------------------------------- /models/atlasnet/ops/cuda/nn_distance_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/cuda/nn_distance_kernel.cu -------------------------------------------------------------------------------- /models/atlasnet/ops/emd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/emd.py -------------------------------------------------------------------------------- /models/atlasnet/ops/fps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/fps.py -------------------------------------------------------------------------------- /models/atlasnet/ops/group_points.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/group_points.py -------------------------------------------------------------------------------- /models/atlasnet/ops/knn_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/knn_distance.py -------------------------------------------------------------------------------- /models/atlasnet/ops/nn_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/nn_distance.py -------------------------------------------------------------------------------- /models/atlasnet/ops/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/setup.py -------------------------------------------------------------------------------- /models/atlasnet/ops/tests/test_ball_query.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/tests/test_ball_query.py -------------------------------------------------------------------------------- /models/atlasnet/ops/tests/test_chamfer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/tests/test_chamfer.py -------------------------------------------------------------------------------- /models/atlasnet/ops/tests/test_fps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/tests/test_fps.py -------------------------------------------------------------------------------- /models/atlasnet/ops/tests/test_knn_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/tests/test_knn_distance.py -------------------------------------------------------------------------------- /models/atlasnet/ops/tests/test_nn_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/atlasnet/ops/tests/test_nn_distance.py -------------------------------------------------------------------------------- /models/base_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/base_decoder.py -------------------------------------------------------------------------------- /models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/base_model.py -------------------------------------------------------------------------------- /models/decoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/decoder/__init__.py -------------------------------------------------------------------------------- /models/decoder/atlas_cycle_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/decoder/atlas_cycle_decoder.py -------------------------------------------------------------------------------- /models/decoder/conv_template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/decoder/conv_template.py -------------------------------------------------------------------------------- /models/decoder/cycle_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/decoder/cycle_decoder.py -------------------------------------------------------------------------------- /models/decoder/geometry_mlp_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/decoder/geometry_mlp_decoder.py -------------------------------------------------------------------------------- /models/decoder/geometry_volume_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/decoder/geometry_volume_decoder.py -------------------------------------------------------------------------------- /models/decoder/original_nerf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/decoder/original_nerf.py -------------------------------------------------------------------------------- /models/decoder/standard_volume_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/decoder/standard_volume_decoder.py -------------------------------------------------------------------------------- /models/diff_ray_marching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/diff_ray_marching.py -------------------------------------------------------------------------------- /models/diff_render_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/diff_render_func.py -------------------------------------------------------------------------------- /models/diff_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/diff_transforms.py -------------------------------------------------------------------------------- /models/embedding/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/embedding/__init__.py -------------------------------------------------------------------------------- /models/embedding/gaussian_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/embedding/gaussian_embedding.py -------------------------------------------------------------------------------- /models/embedding/lp_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/embedding/lp_embedding.py -------------------------------------------------------------------------------- /models/encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/encoders.py -------------------------------------------------------------------------------- /models/nerf_atlas_radiance_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/nerf_atlas_radiance_model.py -------------------------------------------------------------------------------- /models/nerf_decoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/nerf_decoders.py -------------------------------------------------------------------------------- /models/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/networks.py -------------------------------------------------------------------------------- /models/point_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/point_decoder.py -------------------------------------------------------------------------------- /models/texture/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/texture/__init__.py -------------------------------------------------------------------------------- /models/texture/texture_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/texture/texture_mlp.py -------------------------------------------------------------------------------- /models/texture_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/texture_networks.py -------------------------------------------------------------------------------- /models/volumetric_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/models/volumetric_networks.py -------------------------------------------------------------------------------- /options/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/options/__init__.py -------------------------------------------------------------------------------- /options/base_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/options/base_options.py -------------------------------------------------------------------------------- /options/test_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/options/test_options.py -------------------------------------------------------------------------------- /options/train_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/options/train_options.py -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/readme.md -------------------------------------------------------------------------------- /run/DTU/scan114/trainData/data.hdf5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/run/DTU/scan114/trainData/data.hdf5 -------------------------------------------------------------------------------- /run/DTU/scan114/trainData/in_bboxes.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/run/DTU/scan114/trainData/in_bboxes.npy -------------------------------------------------------------------------------- /run/DTU/scan114/trainData/in_camAts.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/run/DTU/scan114/trainData/in_camAts.npy -------------------------------------------------------------------------------- /run/DTU/scan114/trainData/in_camExtrinsics.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/run/DTU/scan114/trainData/in_camExtrinsics.npy -------------------------------------------------------------------------------- /run/DTU/scan114/trainData/in_camFocal.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/run/DTU/scan114/trainData/in_camFocal.npy -------------------------------------------------------------------------------- /run/DTU/scan114/trainData/in_camOrgs.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/run/DTU/scan114/trainData/in_camOrgs.npy -------------------------------------------------------------------------------- /run/DTU/scan114/trainData/in_camPrincpt.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/run/DTU/scan114/trainData/in_camPrincpt.npy -------------------------------------------------------------------------------- /run/DTU/scan114/trainData/in_camUps.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/run/DTU/scan114/trainData/in_camUps.npy -------------------------------------------------------------------------------- /run/DTU/scan114/trainData/mesh_unit.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/run/DTU/scan114/trainData/mesh_unit.ply -------------------------------------------------------------------------------- /run/DTU/scan114/trainData/pcd_down_unit.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/run/DTU/scan114/trainData/pcd_down_unit.ply -------------------------------------------------------------------------------- /run/dtu-freeze.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/run/dtu-freeze.sh -------------------------------------------------------------------------------- /run/dtu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/run/dtu.sh -------------------------------------------------------------------------------- /run/dtu.vis.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/run/dtu.vis.sh -------------------------------------------------------------------------------- /run/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/run/train.py -------------------------------------------------------------------------------- /run/visualize_nerf_atlas_radiance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/run/visualize_nerf_atlas_radiance.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/camera.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/utils/camera.py -------------------------------------------------------------------------------- /utils/cube_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/utils/cube_map.py -------------------------------------------------------------------------------- /utils/format.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/utils/format.py -------------------------------------------------------------------------------- /utils/grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/utils/grid.py -------------------------------------------------------------------------------- /utils/grid_viewer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/utils/grid_viewer.py -------------------------------------------------------------------------------- /utils/icosphere.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/string_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/utils/string_util.py -------------------------------------------------------------------------------- /utils/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fbxiang/NeuTex/HEAD/utils/visualizer.py --------------------------------------------------------------------------------