pix2xray: converting RGB images into X-rays using generative adversarial networks

Int J Comput Assist Radiol Surg. 2020 Jun;15(6):973-980. doi: 10.1007/s11548-020-02159-2. Epub 2020 Apr 27.

Abstract

Purpose: We propose a novel methodology for generating synthetic X-rays from 2D RGB images. This method creates accurate simulations for use in non-diagnostic visualization problems where the only input comes from a generic camera. Traditional methods are restricted to using simulation algorithms on 3D computer models. To solve this problem, we propose a method of synthetic X-ray generation using conditional generative adversarial networks (CGANs).

Methods: We create a custom synthetic X-ray dataset generator to generate image triplets for X-ray images, pose images, and RGB images of natural hand poses sampled from the NYU hand pose dataset. This dataset is used to train two general-purpose CGAN networks, pix2pix and CycleGAN, as well as our novel architecture called pix2xray which expands upon the pix2pix architecture to include the hand pose into the network.

Results: Our results demonstrate that our pix2xray architecture outperforms both pix2pix and CycleGAN in producing higher-quality X-ray images. We measure higher similarity metrics in our approach, with pix2pix coming in second, and CycleGAN producing the worst results. Our network performs better in the difficult cases which involve high occlusion due to occluded poses or large rotations.

Conclusion: Overall our work establishes a baseline that synthetic X-rays can be simulated using 2D RGB input. We establish the need for additional data such as the hand pose to produce clearer results and show that future research must focus on more specialized architectures to improve overall image clarity and structure.

Keywords: Conditional generative adversarial networks; DRR; Image translation; Machine learning; X-rays.

MeSH terms

  • Algorithms
  • Computer Simulation
  • Humans
  • Image Processing, Computer-Assisted / methods*
  • Radiography / methods*
  • X-Rays*