search

qubvel / segmentation_models

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.
MIT License
4.66k stars 1.03k forks source link

ModuleNotFoundError: No module named 'segmentation_models' #543

Open Abecadarian opened 1 year ago

Abecadarian commented 1 year ago

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'

baselmousi commented 1 year ago

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

Alleyaa commented 8 months ago

still not working