Tensorflow Object 检测 API - 错误运行 model_builder_test.py 模块 'tensorflow' 没有属性 'contrib'

Question

I installed the Tensorflow Object Detection API, and ran the model_builder_test.py script to make sure everything was working. I got the following error:

AttributeError: module 'tensorflow' has no attribute 'contrib'

I'm using Python 3.7.3 and Tensorflow 2.0.0. According to this answer , it may be related to Tensorflow version 2. I'm going to use this method to upgrade the model_builder_test.py script. However, I'm worried about other issues in the Object Detection API using Tensorflow 2.

My questions are:

1) Am I correct in interpreting this error?

2) Is it safe to use Object Detection with Tensorflow 2, or should I downgrade to Tensorflow 1.x?

Thanks!

Answer 1

1) Yes

2) Yes, and it may in fact work better per several bug fixes in TF2 - but make sure you follow the linked guide closely to confirm model behavior doesn't change unexpectedly (ie compare execution in TF1 vs. TF2)

However ; the "make sure" in (2) is easier said than done - we're talking about an entire API here. This is best left to the API's devs themselves, unless you're highly familiar with relevant parts of the repository . Even if you fix one bug, there may be others, even those that don't throw errors, per class/method-based functionality changes (especially in Eager vs. Graph interactions). There's not much harm to using TF 1.x, and it may even run faster .

Lastly, I'd suggest opening a TF Git issue on this; contributors/devs may respond there & not here.