对象检测API中对Tensorflow 2.0的支持

时间:2019-04-16 19:14:24

标签: tensorflow tensorflow-serving object-detection-api

我正在使用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'

2 个答案:

答案 0 :(得分:1)

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

答案 1 :(得分:0)

Tensorflow 2.0仍在Alpha中。幸运的是,最终将添加对Tensorflow 2.0的支持。看到这个issue