qubvel / segmentation_models

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ModuleNotFoundError: No module named 'segmentation_models'

Abecadarian opened this issue · comments

When I try to install segmentation_models in Google CoLab, the library appears to install correctly. However, when I try to run it, I get a message saying:

ModuleNotFoundError: No module named 'segmentation_models'

Hello,

Try to add

from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession

config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)

and check whether it works

still not working