R^2-Gaussian: Rectifying Radiative Gaussian Splatting for Tomographic Reconstruction复现

发布于 | 分类于 Python| 0

https://github.com/Ruyi-Zha/r2_gaussian

Visual Studio 2019安装适配nvcc的msvc

使用MSVC 编译器

如何在 Windows + MSVC 环境下开启 POSIX 数学常量支持？

文章目录 隐藏
1 安装与配置
2 训练
3 目录树

安装与配置

We notice that there might be some installation issues with Conda. You can alternatively try installing via pip (adapted from 4DRGS).

git clone https://github.com/Ruyi-Zha/r2_gaussian.git --recursive
conda create -n r2_gaussian python=3.9 -y
conda activate r2_gaussian

pip install torch==2.1.2+cu118 torchvision==0.16.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
pip install -r requirements.txt
pip install -e r2_gaussian/submodules/simple-knn
pip install -e r2_gaussian/submodules/xray-gaussian-rasterization-voxelization

wget https://github.com/CERN/TIGRE/archive/refs/tags/v2.3.zip
unzip v2.3.zip
pip install TIGRE-2.3/Python --no-build-isolation

 

训练

python scripts/train_all.py --source data/synthetic_dataset/cone_ntrain_50_angle_360 --output output/synthetic_dataset/cone_ntrain_50_angle_360 --device 0

 

目录树

Windows PowerShell
版权所有（C） Microsoft Corporation。保留所有权利。

安装最新的 PowerShell，了解新功能和改进！https://aka.ms/PSWindows

加载个人及系统配置文件用了 2905 毫秒。
(base) PS C:\Users\24764\Desktop\r2_gaussian> tree
卷 Windows-SSD 的文件夹 PATH 列表
卷序列号为 BE73-DB1F
C:.
├─.vscode
├─assets
├─data
│ ├─real_dataset
│ │ ├─all_projection
│ │ │ ├─pine
│ │ │ ├─seashell
│ │ │ └─walnut
│ │ ├─cone_ntrain_25_angle_360
│ │ │ ├─pine
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─seashell
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ └─walnut
│ │ │ ├─proj_test
│ │ │ └─proj_train
│ │ ├─cone_ntrain_50_angle_360
│ │ │ ├─pine
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─seashell
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ └─walnut
│ │ │ ├─proj_test
│ │ │ └─proj_train
│ │ └─cone_ntrain_75_angle_360
│ │ ├─pine
│ │ │ ├─proj_test
│ │ │ └─proj_train
│ │ ├─seashell
│ │ │ ├─proj_test
│ │ │ └─proj_train
│ │ └─walnut
│ │ ├─proj_test
│ │ └─proj_train
│ ├─real_dataset_naf_format
│ │ ├─cone_ntrain_25_angle_360
│ │ ├─cone_ntrain_50_angle_360
│ │ └─cone_ntrain_75_angle_360
│ ├─synthetic_dataset
│ │ ├─cone_ntrain_25_angle_360
│ │ │ ├─0_chest_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─0_foot_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─0_head_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─0_jaw_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─0_pancreas_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─1_beetle_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─1_bonsai_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─1_broccoli_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─1_kingsnake_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─1_pepper_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─2_backpack_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─2_engine_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─2_mount_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─2_present_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ └─2_teapot_cone
│ │ │ ├─proj_test
│ │ │ └─proj_train
│ │ ├─cone_ntrain_50_angle_360
│ │ │ ├─0_chest_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─0_foot_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─0_head_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─0_jaw_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─0_pancreas_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─1_beetle_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─1_bonsai_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─1_broccoli_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─1_kingsnake_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─1_pepper_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─2_backpack_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─2_engine_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─2_mount_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ ├─2_present_cone
│ │ │ │ ├─proj_test
│ │ │ │ └─proj_train
│ │ │ └─2_teapot_cone
│ │ │ ├─proj_test
│ │ │ └─proj_train
│ │ └─cone_ntrain_75_angle_360
│ │ ├─0_chest_cone
│ │ │ ├─proj_test
│ │ │ └─proj_train
│ │ ├─0_foot_cone
│ │ │ ├─proj_test
│ │ │ └─proj_train
│ │ ├─0_head_cone
│ │ │ ├─proj_test
│ │ │ └─proj_train
│ │ ├─0_jaw_cone
│ │ │ ├─proj_test
│ │ │ └─proj_train
│ │ ├─0_pancreas_cone
│ │ │ ├─proj_test
│ │ │ └─proj_train
│ │ ├─1_beetle_cone
│ │ │ ├─proj_test
│ │ │ └─proj_train
│ │ ├─1_bonsai_cone
│ │ │ ├─proj_test
│ │ │ └─proj_train
│ │ ├─1_broccoli_cone
│ │ │ ├─proj_test
│ │ │ └─proj_train
│ │ ├─1_kingsnake_cone
│ │ │ ├─proj_test
│ │ │ └─proj_train
│ │ ├─1_pepper_cone
│ │ │ ├─proj_test
│ │ │ └─proj_train
│ │ ├─2_backpack_cone
│ │ │ ├─proj_test
│ │ │ └─proj_train
│ │ ├─2_engine_cone
│ │ │ ├─proj_test
│ │ │ └─proj_train
│ │ ├─2_mount_cone
│ │ │ ├─proj_test
│ │ │ └─proj_train
│ │ ├─2_present_cone
│ │ │ ├─proj_test
│ │ │ └─proj_train
│ │ └─2_teapot_cone
│ │ ├─proj_test
│ │ └─proj_train
│ └─synthetic_dataset_naf_format
│ ├─cone_ntrain_25_angle_360
│ ├─cone_ntrain_50_angle_360
│ └─cone_ntrain_75_angle_360
├─data_generator
│ ├─naf_dataset
│ ├─real_dataset
│ └─synthetic_dataset
│ └─scanner
├─output
│ └─synthetic_dataset
│ ├─cone_ntrain_25_angle_360
│ │ └─0_chest_cone
│ │ ├─ckpt
│ │ └─eval
│ │ ├─iter_000001
│ │ ├─iter_005000
│ │ ├─iter_010000
│ │ └─iter_020000
│ └─cone_ntrain_50_angle_360
│ └─0_chest_cone
│ ├─ckpt
│ └─eval
│ ├─iter_000001
│ └─iter_005000
├─r2_gaussian
│ ├─arguments
│ │ └─__pycache__
│ ├─dataset
│ │ └─__pycache__
│ ├─gaussian
│ │ └─__pycache__
│ ├─submodules
│ │ ├─simple-knn
│ │ │ ├─build
│ │ │ │ ├─lib.win-amd64-cpython-39
│ │ │ │ │ └─simple_knn
│ │ │ │ └─temp.win-amd64-cpython-39
│ │ │ │ └─Release
│ │ │ │ └─simple_knn
│ │ │ ├─simple_knn
│ │ │ └─simple_knn.egg-info
│ │ └─xray-gaussian-rasterization-voxelization
│ │ ├─build
│ │ │ ├─lib.win-amd64-cpython-39
│ │ │ │ └─xray_gaussian_rasterization_voxelization
│ │ │ └─temp.win-amd64-cpython-39
│ │ │ └─Release
│ │ │ ├─cuda_rasterizer
│ │ │ └─cuda_voxelizer
│ │ ├─cuda_rasterizer
│ │ ├─cuda_voxelizer
│ │ ├─third_party
│ │ │ └─glm
│ │ │ ├─.github
│ │ │ │ └─workflows
│ │ │ ├─cmake
│ │ │ ├─doc
│ │ │ │ ├─api
│ │ │ │ │ └─search
│ │ │ │ ├─manual
│ │ │ │ └─theme
│ │ │ ├─glm
│ │ │ │ ├─detail
│ │ │ │ ├─ext
│ │ │ │ ├─gtc
│ │ │ │ ├─gtx
│ │ │ │ └─simd
│ │ │ ├─test
│ │ │ │ ├─bug
│ │ │ │ ├─cmake
│ │ │ │ ├─core
│ │ │ │ ├─ext
│ │ │ │ ├─gtc
│ │ │ │ ├─gtx
│ │ │ │ └─perf
│ │ │ └─util
│ │ ├─xray_gaussian_rasterization_voxelization
│ │ │ └─__pycache__
│ │ └─xray_gaussian_rasterization_voxelization.egg-info
│ └─utils
│ └─__pycache__
├─results
├─scripts
├─TIGRE-2.3
│ ├─.idea
│ ├─Common
│ │ ├─CUDA
│ │ └─data
│ ├─Frontispiece
│ ├─MATLAB
│ │ ├─Algorithms
│ │ ├─Colormaps
│ │ ├─Demos
│ │ ├─Test_data
│ │ │ ├─MRheadbrain
│ │ │ ├─pCT_test_data
│ │ │ └─Shepp_logan
│ │ ├─Third_party_tools
│ │ │ ├─arrow3d
│ │ │ ├─readMHD
│ │ │ └─sec2hours
│ │ └─Utilities
│ │ ├─cuda_interface
│ │ ├─GPU
│ │ ├─IO
│ │ │ ├─Bruker
│ │ │ ├─Dicom
│ │ │ ├─Nikon
│ │ │ └─VarianCBCT
│ │ │ └─polar2cart
│ │ ├─Quality_measures
│ │ └─Setup
│ └─Python
│ ├─build
│ │ ├─bdist.linux-x86_64
│ │ ├─bdist.win-amd64
│ │ ├─Common
│ │ │ └─CUDA
│ │ ├─lib.linux-x86_64-cpython-39
│ │ │ └─tigre
│ │ │ ├─algorithms
│ │ │ └─utilities
│ │ │ ├─cuda_interface
│ │ │ ├─io
│ │ │ └─visualization
│ │ ├─lib.win-amd64-cpython-39
│ │ │ └─tigre
│ │ │ ├─algorithms
│ │ │ └─utilities
│ │ │ ├─cuda_interface
│ │ │ ├─io
│ │ │ └─visualization
│ │ ├─temp.linux-x86_64-cpython-39
│ │ │ └─tigre
│ │ │ └─utilities
│ │ │ └─cuda_interface
│ │ └─temp.win-amd64-cpython-39
│ │ ├─Common
│ │ │ └─CUDA
│ │ └─Release
│ │ └─tigre
│ │ └─utilities
│ │ └─cuda_interface
│ ├─demos
│ ├─pytigre.egg-info
│ ├─tests
│ └─tigre
│ ├─algorithms
│ ├─data
│ └─utilities
│ ├─cuda_interface
│ ├─io
│ └─visualization
└─torch_wheels
(base) PS C:\Users\24764\Desktop\r2_gaussian>

作者： 高志远

高志远，24岁，男生

发表评论

邮箱地址不会被公开。