前提・実現したいこと
自前モデルを使ってyoloを実装したい
yolov3を使って、車の車種をアノテしたモデルを作成しています。
wget https://pjreddie.com/media/files/yolov3.weights
でyoloの重みをwegetし、下記のコードを実装した際にエラーが発生します。
python convert.py -w yolov3.cfg yolov3.weights model_data/yolo_weights.h5
tf version : tensorflow 2.0 -GPU
発生している問題・エラーメッセージ
2019-10-06 14:34:17.735282: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_100.dll Using TensorFlow backend. Loading weights. Weights Header: 0 2 0 [32013312] Parsing Darknet config. Creating Keras model. Traceback (most recent call last): File "convert.py", line 259, in <module> _main(parser.parse_args()) File "convert.py", line 85, in _main input_layer = Input(shape=(None, None, 3)) File "C:\Users\skawa\AppData\Local\Continuum\anaconda3\envs\my\lib\site-packages\keras\engine\input_layer.py", line 178, in Input input_tensor=tensor) File "C:\Users\skawa\AppData\Local\Continuum\anaconda3\envs\my\lib\site-packages\keras\legacy\interfaces.py", line 91, in wrapper return func(*args, **kwargs) File "C:\Users\skawa\AppData\Local\Continuum\anaconda3\envs\my\lib\site-packages\keras\engine\input_layer.py", line 39, in __init__ name = prefix + '_' + str(K.get_uid(prefix)) File "C:\Users\skawa\AppData\Local\Continuum\anaconda3\envs\my\lib\site-packages\keras\backend\tensorflow_backend.py", line 74, in get_uid graph = tf.get_default_graph() AttributeError: module 'tensorflow' has no attribute 'get_default_graph'
該当のソースコード
python
1from tensorflow.python.keras import backend as K
試したこと
gitやHP上で調べてみたところ、
from tensorflow.python.keras import backend as K
の部分を
from tensorflow.keras import backend as Kとしても
変わらずにエラーが出てしまっています。
補足情報(FW/ツールのバージョンなど)
tensorflow-gpu 2.0.0
Keras 2.2.4
ご教授をいただけませんでしょうか?
よろしくお願い申し上げます!
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。