This is the official pytorch implementation of "VLD-Net: Localization and Detection of the Vertebrae from X-ray Images by Reinforcement Learning with Adaptive Exploration Mechanism and Spine Anatomy Information"
CUDA 11.4
Python 3.7
Tensorflow-gpu 2.5.0
Tensorboard 2.11.0
- Clone this repo
git clone https://github.com/hlyf-xs/VLD-Net.git
cd VLD-Net
-
Download MICCAI 2019 AASCE Challenge Dataset, and BUU LSPINE dataset, then put them into
data/ -
Train the model
sh run.sh
If you find our work is useful for you, please cite us.
Shun Xiang ([email protected])