1. ホーム
  2. TensorFlow

Keras-Yolo v3 のエラーを解決する。AttributeError: モジュール 'keras.backend' には 'control_flow_ops' という属性がありません。

2022-02-20 20:52:48
<パス
  • モジュール 'keras.backend' は、Keras-YOLO v3 で独自のデータセットをトレーニングする際、属性 'control_flow_ops' がなく、次のエラーメッセージが表示されます。
  • エラーメッセージを表示します。
 Freeze the first 249 layers of total 252 layers.
Traceback (most recent call last):
  File "train.py", line 195, in <module>
    _main()
  File "train.py", line 38, in _main
    freeze_body=2, weights_path='model_data/yolo_weights.h5') # make sure you know what you freeze
  File "train.py", line 135, in create_model
    [*model_body.output, *y_true])
  File "/home/ligy/anaconda3/lib/python3.6/site-packages/Keras-2.2.4-py3.6.egg/keras/engine/base_layer.py", line 457, in __call __
    output = self.call(inputs, **kwargs)
  File "/home/ligy/anaconda3/lib/python3.6/site-packages/Keras-2.2.4-py3.6.egg/keras/layers/core.py", line 716, in call
    return self.function(inputs, **arguments)
  File "/home/ligy/Programming/ShellDetection/keras/keras-yolo3-master/yolo3/model.py", line 394, in yolo_loss
    _, ignore_mask = K.control_flow_ops.while_loop(lambda b,*args: b<m, loop_body, [0, ignore_mask])
AttributeError: module 'keras.backend' has no attribute 'control_flow_ops'



  • 解決策
  • 以下のディレクトリにある "__ init__.py" ファイルを修正してください、みんなのパスが同じではないので、自分で修正してください〜。
# Open the directory file
vim ~/anaconda3/lib/python3.6/site-packages/Keras-2.2.4-py3.6.egg/keras/backend/__init__.py

# Add the following package import statement at around line 150
from .load_backend import control_flow_ops
from .load_backend import set_image_dim_ordering

# Save and close
Key sequence: ESC --> : w q Enter (English state, : indispensable)


図に示すように

もう一度トレーニングプログラムを実行すると、エラーが報告されなくなり、問題なく実行できるようになりました〜。