4 个月前

结构感知稀疏视图X射线3D重建

结构感知稀疏视图X射线3D重建

摘要

X射线以其能够揭示物体内部结构的能力而著称,预计在三维重建中比可见光提供更丰富的信息。然而,现有的神经辐射场(NeRF)算法忽视了X射线这一重要特性,导致其在捕捉成像物体的结构内容方面存在局限性。本文提出了一种框架——结构感知X射线神经辐射密度场(SAX-NeRF),用于稀疏视角下的X射线三维重建。首先,我们设计了一种基于线段的Transformer(Lineformer)作为SAX-NeRF的骨干网络。Lineformer通过建模每个X射线线段内的依赖关系来捕捉物体在三维空间中的内部结构。其次,我们提出了一种掩码局部-全局(MLG)光线采样策略,以提取二维投影中的上下文和几何信息。此外,我们收集了一个更大规模的数据集X3D,涵盖了更广泛的X射线应用。在X3D数据集上的实验表明,SAX-NeRF在新视图合成和CT重建方面分别超越了之前的NeRF方法12.56 dB和2.49 dB。代码、模型和数据已发布在https://github.com/caiyuanhao1998/SAX-NeRF

代码仓库

基准测试

基准方法指标
low-dose-x-ray-ct-reconstruction-on-x3dASD-POCS
PSNR: 32.32
SSIM: 0.9400
low-dose-x-ray-ct-reconstruction-on-x3dSAX-NeRF
PSNR: 37.25
SSIM: 0.9753
low-dose-x-ray-ct-reconstruction-on-x3dSART
PSNR: 32.33
SSIM: 0.9342
low-dose-x-ray-ct-reconstruction-on-x3dFDK
PSNR: 25.12
SSIM: 0.6422
novel-view-synthesis-on-x3dSAX-NeRF
PSNR: 51.37
SSIM: 0.9994

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
结构感知稀疏视图X射线3D重建 | 论文 | HyperAI超神经