GitHub - qubvel/segmentation_models: Segmentation models with ...
github.com › qubvel › segmentation_modelsApr 17, 2020 · Provide environment variable SM_FRAMEWORK=keras / SM_FRAMEWORK=tf.keras before import segmentation_models; Change framework sm.set_framework('keras') / sm.set_framework('tf.keras') You can also specify what kind of image_data_format to use, segmentation-models works with both: channels_last and channels_first. This can be useful for further model conversion to Nvidia TensorRT format or optimizing model for cpu/gpu computations.