tensorflow-yolov3 icon indicating copy to clipboard operation
tensorflow-yolov3 copied to clipboard

按照readme安装好环境,执行convert_weight.py报错,请各位高手指教,谢谢!

Open junlintianxiatjm opened this issue 4 years ago • 1 comments

2021-01-20 17:01:33.778610: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine. Traceback (most recent call last): File "D:/python-workspace/tensorflow-yolov3-master/convert_weight.py", line 30, in load = tf.train.import_meta_graph(org_weights_path + '.meta') AttributeError: module 'tensorflow._api.v2.train' has no attribute 'import_meta_graph'

junlintianxiatjm avatar Jan 20 '21 09:01 junlintianxiatjm

tensorflow版本过高,降低版本就行了: #import tensorflow as tf import tensorflow.compat.v1 as tf tf.compat.v1.disable_eager_execution()

Atange1998 avatar Jan 30 '21 03:01 Atange1998