在 Object Detection API 中支持 Tensorflow 2.0

Question

我正在使用 TF 对象检测 API 来训练我最终将使用 tf-serving 部署的模型。我计划获取这个网络的输出（在中间 CNN 层）并为除对象检测之外的其他功能构建额外的网络。我计划在 TF 2.0 中开发新的网络。然而，为了让这个工作顺利进行，（我怀疑）需要更新对象检测 API 以支持 TF-2.0。

是否有计划将 TF OD API 更新到 TF 2.0？

我尝试运行 model_builder_test 失败，如下所示，因为 tensorflow 2.0 不再支持 contrib。

Traceback (most recent call last):
  File "object_detection/builders/model_builder_test.py", line 23, in <module>
    from object_detection.builders import model_builder
  File "/models/research/object_detection/builders/model_builder.py", line 19, in <module>
    from object_detection.builders import box_predictor_builder
  File "/models/research/object_detection/builders/box_predictor_builder.py", line 18, in <module>
    from object_detection.core import box_predictor
  File "/research/object_detection/core/box_predictor.py", line 35, in <module>
    slim = tf.contrib.slim
AttributeError: module 'tensorflow' has no attribute 'contrib'

Answer 1

Tensorflow 2.0 仍处于 Alpha 阶段。幸运的是，最终会添加对 TensorFlow 2.0 的支持。看到这个issue

Answer 2

是的，有计划将对象检测 API 迁移到 TF2.0，但需要几个月的时间。同时请使用 TF1.x 作为 API。