caiyuanhao1998 / SAX-NeRF

"Structure-Aware Sparse-View X-ray 3D Reconstruction" (CVPR 2024)
https://arxiv.org/abs/2311.10959
MIT License
467 stars 19 forks source link

One view 2D x-ray image to 3D images #18

Closed dicaleon closed 2 months ago

dicaleon commented 2 months ago

Hello. Thank you for your great job! I have a question for your code.

I try to convert a 2D x-ray image (only one view) to 3D images. I just want to know if your code has this function or not. If it has, then, do the training data also only require only one view 2D x-ray image?

caiyuanhao1998 commented 2 months ago

Hi, thanks for your interest.

I have not tried single-view reconstruction. But I think it is interesting to do that.

Just take a single projection in the pickle data to train by modifying the loaded data here:

https://github.com/caiyuanhao1998/SAX-NeRF/blob/master/src/dataset/tigre_mlg.py#L75

If you find our repo useful, please help us star it, thanks!