无法 运行 model_main.py 训练数据。贡献错误

Unable to run model_main.py for training data. contrib error

系统信息

Windows 10 TensorFlow 版本 2.0.0 Python虚拟环境中的3.6.9 CUDA/cuDNN:是的 显卡型号和内存:GTX1080Ti,11GB

我试过 运行 python main_model.py --logtostderr --train_dir=training/ --pipeline_config_path=training/faster_rcnn_inception_v2_pets.config

但由于属性错误无法这样做。这是显示的错误:

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

我对 https://github.com/tensorflow/tensorflow/issues/33442 做了一些研究 我能得到的最好的被告知使用这个 link 找到替代方案:https://github.com/tensorflow/addons

老实说,我对此还是个新手,不知道用什么来替换 tensorflow 2.0.0 中的贡献

我目前正在学习 https://github.com/EdjeElectronics/TensorFlow-Object-Detection-API-Tutorial-Train-Multiple-Objects-Windows-10

中的教程

实际结果应该可以训练数据无误

由于 tf.contrib 中的所有项目都没有得到 Tensorflow 的官方支持,并且它有指定的所有者来维护它。 所有贡献和功能都旨在合并到核心 Tensorflow 中。

从 Tensorflow 2.0 版本开始,所有 contrib 项目的未来都有以下三种选择之一:move to coremove to a separate repository;或删除,其中大部分已与各自的项目所有者一起审核。

link 提供 tf.contrib 下所有项目的详细信息和状态。

如果您正在使用的库被移动到核心或单独的存储库,TensorFlow 的自动代码从 1.x 迁移到 2.x 将不适用于 tf.contrib projects。您需要手动更改这些部分的代码,建议继续这样做。

希望这能回答您的问题,祝您学习愉快!