This project provides a GUI for chest X-ray (CXR) inspiration and rotation inspection. π©»β¨
The WYVERN application uses deep learning models to analyze chest X-ray images. Through its graphical user interface, you can:
- Click "Select Img" to choose images (.dcm or .png or .jpg) for inspection
- Click "Analyze" button to start the process
- We detect inspiration by segmenting the lungs and 9th posterior rib and calculate overlap ratio called 'Rib over Lung (ROL)' to determine if the inspiration is sufficient. If it above a certain threshold, we classify it as 'Adequate Inspiration', otherwise 'Inadequate Inspiration'.
- For rotation, we analyze the x position medial end of clavicles and the spinous process to determine if the patient is rotated. We calculate 'Alpha' which is quantify metric for access symmetry. If the alpha is above a certain threshold, we classify it as 'Rotated', otherwise 'Not Rotated'.
The models in src/models are used to perform the image analysis, ensuring accurate and fast inspection. The GUI makes it easy for users to operate without needing to write code.
-
π¦ Install the required Python packages:
pip install -r requirements.txt
-
π§ Place the required model files in the
src/modelsdirectory. -
π Run the main application:
python main.py
βββ README.md
βββ requirements.txt
βββ main.py # main application file for the GUI (runs the application)
βββ src/
β βββ logo/
| | βββ Logo.png
| | βββ icon.png
| | βββ icon.ico
| β βββ demo.png
β βββ models/
β βββ lung_segmentation_model.pth # pre-trained model for lung segmentation
β βββ rib_segmentation_model.pth # pre-trained model for rib segmentation
β βββ rotation_model.pth # pre-trained model for rotation detection
βββ result/
| βββ result.png # result image will be saved here
βββ example_images/ # example images for testing
βββ EX166/ # DICOM images
βββ artifact/ # images with artifacts
βββ not full/ # images with inadequate inspiration
βββ full/ # images with adequate inspiration
