モデルをトレーニングしようとしたときにエラーが発生しました:
(tensorflow1) C:\tensorflow1\models\research\object_detection>python train.py --logtostderr --train_dir=training/ --pipeline_config_path=training/faster_rcnn_inception_v2_pets.config
Traceback (most recent call last):
File "train.py", line 52, in <module>
from object_detection.builders import model_builder
File "C:\tensorflow1\models\research\object_detection\builders\model_builder.py", line 35, in <module>
from object_detection.models import faster_rcnn_inception_resnet_v2_feature_extractor as frcnn_inc_res
File "C:\tensorflow1\models\research\object_detection\models\faster_rcnn_inception_resnet_v2_feature_extractor.py", line 30, in <module>
from nets import inception_resnet_v2
File "C:\tensorflow1\models\research\slim\nets\inception_resnet_v2.py", line 375, in <module>
batch_norm_updates_collections=tf.compat.v1.GraphKeys.UPDATE_OPS,
AttributeError: module 'tensorflow.compat' has no attribute 'v1'
Tensorflowバージョン:1.10.0
私は次で作成されたconda virtualenvを使用しています:
conda create -n tensorflow1 pip python=3.5
(tensorflow1) conda install tensorflow-gpu
Windows 10での作業
コミュニティの利益のためにコメントセクションに存在する場合でも、ここ(回答セクション)でソリューションを提供します。
このコードtf.compat.v1.GraphKeys.UPDATE_OPS
はTensorflow==1.10.0
では利用できません
TensorFlowの古いバージョンを使用している場合、 here は、使用するリポジトリのGitHubコミットを示す表です。
これらの機能を使用するには、次のようにバージョンを1.15
または2.2
に更新します
conda install tensorflow=1.15
または
conda install tensorflow=2.2