Open Rakshith2202 opened 1 year ago
Try to add
import os
os.environ['SM_FRAMEWORK'] = 'tf.keras'
before importing keras and segmentation_models.
Adding on to @jensleitloff 's answer, this answer on StackOverflow provides some more alternate approaches.
The accepted answer worked for me on a local machine and the answer in this thread worked for Kaggle/Colab.
this answer helped alot
this may help https://github.com/qubvel/segmentation_models/pull/582
This is the error I am getting when I am trying to import segmentation_models