This repository contains the backend api for our undergraduate thesis project entitled: "ScolioVis: Automated Cobb Angle Measurement on Anterior-Posterior Spine X-Rays using Multi-Instance Keypoint Detection with Keypoint RCNN".
A live demo is available on https://scoliovis.app
For more information on the whole project go to blankeos/scoliovis.
- Python
- FastAPI
- OpenCV
- PyTorch
-
Clone repo
> git clone https://github.com/blankeos/scoliovis-api.git > cd scoliovis-api
-
Create a virtual environment
> python -m venv venv
-
Activate virtual environment
> venv\Scripts\activate # windows > source venv/Scripts/activate # bash/mac
-
Install dependencies
> pip install -r requirements.txt
-
Download the model keypointsrcnn_weights.pt and put inside /models
- Download here: scoliovis-training/releases/keypointsrcnn_weights.pt
-
Run the server
> uvicorn main:app