dpwe commented 3 years ago

Prerequisites

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

1. The entire URL of the file you are using

https://github.com/tensorflow/models/tree/master/research/audioset/yamnet/export.py

2. Describe the bug

Running the export.py example code on MacOS 11.5.2, it fails when attempting to verify the tflite export. Seemingly, instead of pulling the scores output from from the model, it is instead retrieving the embeddings output (1024 dimensions instead of 521).

3. Steps to reproduce

$ python3 export.py yamnet.h5 export
=====
Tue Sep  7 08:34:12 2021 | Building and checking TF2 Module ...
=====
[...]
=====
Tue Sep  7 08:35:00 2021 | Checking TF-Lite model ...
=====

Traceback (most recent call last):
  File "/Users/dpwe/google/models/research/audioset/yamnet/export.py", line 213, in <module>
    main(sys.argv[1:])
  File "/Users/dpwe/google/models/research/audioset/yamnet/export.py", line 207, in main
    tflite_saved_model_dir = make_tflite_export(weights_path, tflite_export_dir)
  File "/Users/dpwe/google/models/research/audioset/yamnet/export.py", line 179, in make_tflite_export
    check_model(run_model, 'yamnet_class_map.csv', params)
  File "/Users/dpwe/google/models/research/audioset/yamnet/export.py", line 78, in check_model
    clip_test(
  File "/Users/dpwe/google/models/research/audioset/yamnet/export.py", line 72, in clip_test
    top_n_class_names = yamnet_classes[top_n_indices]
IndexError: index 719 is out of bounds for axis 0 with size 521

4. Expected behavior

check_model should succeed and execution should continue through completion.

5. Additional context

none

6. System information

MacOS 11.5.2
TensorFlow installed from pip3, pip3/python installed with HomeBrew
TensorFlow version: v2.6.0-rc2-32-g919f693420e 2.6.0
Python version: Python 3.9.6

google-ml-butler[bot] commented 3 years ago

Are you satisfied with the resolution of your issue? Yes No