This repo provides a trained model to seperate the chest x-ray into two views: front view (PA or posterior-anterior) and side view (LL or latero-lateral).
Sometimes when you download the chest x-rays from the internet, there's no text description or meta data associated with the image. If you want to only work with images from a single view, you'll have to manually seperate the images which is tedious. Here I trained a model based on resnet-50 to automate the process and hopefully it can help people facing the same issue.
- Linux or OSX
- NVIDIA GPU
- Python 2.7
- PyTorch v0.2.0
- Numpy
- Install PyTorh and the other dependencies
- Clone this repo:
git clone https://github.com/xinario/chestViewSplit.git
cd chestViewSplit- Download the trained model from this link and put it into folder ./models (mkdir ./models)
- Make sure all your chest x-rays resides in the same folder, e.g. ./chest_xray and then run the following command (-i specifies the input folder and -o specifies the output folder).
python xray_split.py -i chest_xray/ -o chest_xray/- After running the above script, the images from two views will be saved in folder ./chest_xray/front and ./chest_xray/side respectively.
Part of the code are borrowed from fine-tuning.pytorch