Open nattu22 opened 3 years ago
In retinanet issues, someone routed the these version. Installed it, but not working !pip install -U -q segmentation-models !pip install -q tensorflow==2.1 !pip install -q keras==2.3.1 !pip install -q tensorflow-estimator==2.1.
import os os.environ['CUDA_VISIBLE_DEVICES'] = '0' os.environ["SM_FRAMEWORK"] = "tf.keras"
from tensorflow import keras import segmentation_models as sm
It's very old code. It works with TF 1.* version:
Try this settings:
tensorflow-gpu==1.15.3
keras==2.2.3
keras-resnet==0.1.0
keras-retinanet==0.4.1
or this - I used in my latest project with given code. But I probably changed something:
tensorflow-gpu==1.15.3
keras==2.3.1
keras-resnet==0.1.0
keras-retinanet==0.5.1
Hi, I tried with various versions of tensorflow and keras, but not able to run the code. If possible add the requirements.txt.