README in progress, for now please refer to
This project uses machine learning to identify malnutrition from 3D scans of children under 5 years of age. This one-minute video explains.
Training the models realistically requires using GPU computation. Project members are currently using a variety of cloud instances (GCS, AWS, Azure) and local machines for training. A separate backend project is currently developing the DevOps infrastructure to simplify this.
You will need:
- Python 3
- TensorFlow GPU
- Keras
These steps provide an example installation on a local Ubuntu workstation from scratch:
- Install Ubuntu Desktop 18.04.1 LTS
- Install NVIDIA drivers
Please note that after rebooting, the secure boot process will prompt you to authorize the driver to use the hardware via a MOK Management screen.
sudo add-apt-repository ppa:graphics-drivers
sudo apt-get update
sudo apt-get install nvidia-390
sudo reboot now
- Install Anaconda with Python 3.6
conda update anaconda
conda update python
conda update --all
conda create --name cgm
source activate cgm
conda install tensorflow-gpu
conda install ipykernel
conda install keras
conda install vtk progressbar2 glob2 numbs pandas
pip install --upgrade pip
pip install git+https://github.com/daavoo/pyntcloud
Data access is provided on as-needed basis following signature of the Welthungerhilfe Data Privacy & Commitment to Maintain Data Secrecy Agreement. If you need data access (e.g. to train your machine learning models), please contact Markus Matiaschek for details.
- Run
python create_pickle.py
- Run the notebook
create_dataset.ipynb
- Run the notebook
train_on_dataset.ipynb
This currently takes around 6 hours to run on a local NVIDIA GTX 1080 Ti.
TODO
Please see CONTRIBUTING.md for details.
Our releases use semantic versioning. You can find a chronologically ordered list of notable changes in CHANGELOG.md.
This project is licensed under the GNU General Public License v3.0. See LICENSE for details and refer to NOTICE for additional licensing notes and use of third-party components.