Installation

Requirements

1. Python 3
2. Keras >= 2.2.0 or TensorFlow >= 1.13
3. keras-applications >= 1.0.7, <=1.0.8
4. image-classifiers == 1.0.0
5. efficientnet == 1.0.0

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