Cartucho / OpenLabeling

Label images and video for Computer Vision applications
Apache License 2.0
926 stars 266 forks source link

Tensorflow Version #71

Open OzgunKB opened 4 years ago

OzgunKB commented 4 years ago

Error "module 'tensorflow' has no attribute 'GraphDef'"

I cannot run "main_auto.py". Which version of tensorflow should I install?

ark- commented 4 years ago

Also getting this issue.

Pip list output:

Package                Version
---------------------- -----------
absl-py                0.9.0
astunparse             1.6.3
cachetools             4.1.0
certifi                2020.4.5.1
chardet                3.0.4
future                 0.18.2
gast                   0.3.3
google-auth            1.16.0
google-auth-oauthlib   0.4.1
google-pasta           0.2.0
grpcio                 1.29.0
h5py                   2.10.0
idna                   2.9
importlib-metadata     1.6.0
Keras-Preprocessing    1.1.2
lxml                   4.3.0
Markdown               3.2.2
numpy                  1.16.0
oauthlib               3.1.0
opencv-contrib-python  3.4.9.33
opt-einsum             3.2.1
Pillow                 7.1.2
pip                    20.0.2
protobuf               3.12.2
pyasn1                 0.4.8
pyasn1-modules         0.2.8
requests               2.23.0
requests-oauthlib      1.3.0
rsa                    4.0
scipy                  1.4.1
setuptools             45.2.0
six                    1.15.0
tensorboard            2.2.2
tensorboard-plugin-wit 1.6.0.post3
tensorflow             2.2.0
tensorflow-estimator   2.2.0
termcolor              1.1.0
torch                  1.5.0+cu101
torchvision            0.6.0+cu101
tqdm                   4.29.1
urllib3                1.25.9
Werkzeug               1.0.1
wheel                  0.34.2
wrapt                  1.12.1
zipp                   3.1.0

I installed torch with this command as I have CUDA 10.1

pip install torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

I installed tensorflow by using pip install tensorflow.

Hope the additional info helps.

WillJStone commented 4 years ago

tf.GraphDef was removed in tensorflow 2.0

I personally use tensorflow 1.14 when using this tool and have had no issues. It may work with other versions (1.x) as well but I have not tested them personally.

ark- commented 4 years ago

Fantastic, I removed 2.0 and used pip install tensorflow==1.15 (This was the latest 1.x listed on https://www.tensorflow.org/install/pip) . It is working now.