tf-1.5 или tf-2.1 при запуске обучения обнаружению объектов

Question

Когда я пытался запустить:

Из каталога tensorflow / models / research /

PIPELINE_CONFIG_PATH=/home/paul/objdetect/models/research/object_detection/samples/configs/ssdlite_mobilenet_v3_small_320x320_kitti.config
MODEL_DIR=/home/paul/objdetect/models/research/object_detection/samples/sample_model_ckpt/
NUM_TRAIN_STEPS=5000
SAMPLE_1_OF_N_EVAL_EXAMPLES=1
python object_detection/model_main.py \
    --pipeline_config_path=${PIPELINE_CONFIG_PATH} \
    --model_dir=${MODEL_DIR} \
    --num_train_steps=${NUM_TRAIN_STEPS} \
    --sample_1_of_n_eval_examples=$SAMPLE_1_OF_N_EVAL_EXAMPLES \
    --alsologtostderr

, если я использую среду tenorflow-gpu == 1.5.0, я получил следующую ошибку :

    from nets import inception_resnet_v2
  File "/home/paul/objdetect/models/research/slim/nets/inception_resnet_v2.py", line 375, in <module>
    batch_norm_updates_collections=tf.compat.v1.GraphKeys.UPDATE_OPS,
AttributeError: 'module' object has no attribute 'v1'

Затем я переключаюсь на среду tenorflow-gpu == 2.1.0 и снова запускаю ее, у меня появляется следующая ошибка:

    from object_detection import eval_util
  File "/home/paul/objdetect/models/research/object_detection/eval_util.py", line 40, in <module>
    slim = tf.contrib.slim
AttributeError: module 'tensorflow' has no attribute 'contrib'

Так есть ли способ запустить его вообще? что мне нужно исправить, чтобы он работал в tf-1.5 или tf-2.1?

Кто-нибудь может помочь? Заранее большое спасибо.

С уважением,

-Пол

Answer 1

tf.contrib устарел в tf2.x. Аналогично tf.compat был введен в 1.14.x и 1.15.x для решения проблемы совместимости между tf 1.x и tf2.x. Попробуйте установить tenorflow версии 1.15.x и запустить