Requirements
- Python 3.5+
- Keras >= 2.2.0
- Keras Applications >= 1.7.0
- Image Classifiers == 0.2.0
- Tensorflow 1.9 (tested)
Note
This library does not have Tensorflow in a requirements.txt for installation. Please, choose suitable version ('cpu'/'gpu') and install it manually using official Guide.
Pip package
$ pip install segmentation-models
Latest version
$ pip install git+https://github.com/qubvel/segmentation_models