Skip to content

MobileNet's example in google colab does not seem to be working #10161

@Eths33

Description

@Eths33

Prerequisites

Please answer the following questions for yourself before submitting an issue.

  • [Google Colab link ] I am using the latest TensorFlow Model Garden release and TensorFlow 2.
  • [ Yes] I am reporting the issue to the correct repository. (Model Garden official or research directory)
  • [ Tried searching] I checked to make sure that this issue has not already been filed.

1. The entire URL of the file you are using

https://colab.research.google.com/github/tensorflow/models/blob/master/research/slim/nets/mobilenet/mobilenet_example.ipynb

2. Describe the bug

In the section: Checkpoint based inference
The first cell fails to run.

---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
<ipython-input-5-2880d52ed4d2> in <module>()
      1 import tensorflow.compat.v1 as tf
----> 2 from nets.mobilenet import mobilenet_v2
      3 
      4 tf.reset_default_graph()
      5 

/content/models/research/slim/nets/mobilenet/mobilenet_v2.py in <module>()
     29 
     30 import tensorflow.compat.v1 as tf
---> 31 import tf_slim as slim
     32 
     33 from nets.mobilenet import conv_blocks as ops

ModuleNotFoundError: No module named 'tf_slim'
---------------------------------------------------------------------------

3. Steps to reproduce

Click google colab link and start running cells, starting from Prerequisites section.

4. Expected behavior

A working example

5. Additional context

Running !pip install tf_slim fixed the error above but other errors followed. It seems that this was made for TensorFlow version 1 and was not upgraded to version 2 successfully?
The issues that followed included one about eager execution and "module 'tensorflow.compat.v1' has no attribute 'contrib'.

Locally I tried TF version 1 & 2 and had various issues, but my point of posting this issue is to try to request a working colab example.

6. System information

Free googlecolab link.
Tesorflow version 2.5.0

Metadata

Metadata

Labels

models:researchmodels that come under research directorystat:awaiting responseWaiting on input from the contributortype:bugBug in the code

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions