我正在使用TF Object Detection API来训练我最终将使用tf-serving部署的模型。我计划获取此网络的输出(在中间CNN层)并为对象检测之外的其他功能构建其他网络。我打算在TF 2.0中开发新的网络。但是,为了顺利工作,(我怀疑)需要更新Object Detection 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'
Tensorflow 2.0仍处于Alpha状态。幸运的是,最终将添加对Tensorflow 2.0的支持。看到这个issue