module 'keras.utils.generic_utils' has no attribute 'get_custom_objects'

[Rakshith2202]

trafficstars

This is the error I am getting when I am trying to import segmentation_models

[jensleitloff]

Try to add

import os
os.environ['SM_FRAMEWORK'] = 'tf.keras'

before importing keras and segmentation_models.

[vijay-jaisankar]

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.

[sohailjamal]

this answer helped alot

[idhamari]

this may help https://github.com/qubvel/segmentation_models/pull/582