Обнаружение объекта Tensorflow: больше не работают ни model_main.py, ни train.py - PullRequest
Новая
       1

Обнаружение объекта Tensorflow: больше не работают ни model_main.py, ни train.py

1 голос
/ 17 марта 2020

train.py 

Traceback (most recent call last):
  File "object_detection/legacy/train.py", line 48, in <module>
    from tensorflow.contrib import framework as contrib_framework
ModuleNotFoundError: No module named 'tensorflow.contrib'

module_main.py 

    Traceback (most recent call last):
  File "object_detection/model_main.py", line 26, in <module>
    from object_detection import model_lib
  File "/content/models/research/object_detection/model_lib.py", line 27, in <module>
    from object_detection import eval_util
  File "/content/models/research/object_detection/eval_util.py", line 40, in <module>
    slim = tf.contrib.slim
AttributeError: module 'tensorflow' has no attribute 'contrib'

Я даже не могу экспортировать свою модель из контрольной точки в граф вывода из-за одной и той же ошибки.

export_inference_graph.py 

Traceback (most recent call last):
  File "object_detection/export_inference_graph.py", line 108, in <module>
    from object_detection import exporter
  File "/content/models/research/object_detection/exporter.py", line 20, in <module>
    from tensorflow.contrib.quantize.python import graph_matcher
ModuleNotFoundError: No module named 'tensorflow.contrib'

Все три файла работали отлично до сегодняшнего дня. Обучение работало с TF 1.15 и 2.1.0. Я использовал Colab с Python 3.6 и тренировками на GPU. Я также не смог найти каких-либо существенных изменений в последнее время в GitHub репо тензорного потока.

Есть идеи, ребята?

...