This project allows to train, package and deploy a model for pneumonia classification. The models are trained on the RSNA dataset.
Note: this project was not unit tested. Even if it was shown to work, we do not guarantee its usage.
Create a conda environment and type pip install -e .
We offer some helper script to prepare the dataset:
The entry point to the training script is here. You can specify the parameters in your own config file. We provide an example here.
From your conda environment, follow the following steps.
- To package the model in a self contained folder:
python driver_scripts/main_package_model.py -c Capstone/configs/config_RSNA.yml - To create a docker container:
docker build -t flask_api - To serve the container:
docker run -it --rm --gpus all --name flask_cont -p 5000:5000 flask_api