ZFTurbo / Keras-RetinaNet-for-Open-Images-Challenge-2018

Code for 15th place in Kaggle Google AI Open Images - Object Detection Track
MIT License
268 stars 76 forks source link

Tensorflow & Keras version issues #26

Open nattu22 opened 3 years ago

nattu22 commented 3 years ago

Hi, I tried with various versions of tensorflow and keras, but not able to run the code. If possible add the requirements.txt.

nattu22 commented 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

ZFTurbo commented 3 years ago

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