segmentation_models_3D

segmentation_models_3D copied to clipboard

Hi, I work in google colab, try to import segmentation_models_3D as sm, but I get an ImportError: cannot import name 'base' from 'segmentation_models_3D'. I installed keras >= 2.2.0, tensorflow >= 1.13.1, keras_applications >= 1.0.7, classification_models_3D, efficientnet_3D, segmentation_models_3D. What should I do to solve my problem?

Please try to install the following versions:

tensorflow==2.5.0 and keras==2.4.3

or if you need TF1 then:

tensorflow==1.15.0 and keras==2.3.1

I check it worked on them.

Also on TF2 please add the following line after import:

sm.set_framework('tf.keras')

Thanks, it works!