Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Switched YAMNet to stick with Keras 2. #11261

Merged
merged 1 commit into from
Sep 24, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 3 additions & 1 deletion research/audioset/yamnet/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -15,11 +15,13 @@ YAMNet depends on the following Python packages:
* [`numpy`](http://www.numpy.org/)
* [`resampy`](http://resampy.readthedocs.io/en/latest/)
* [`tensorflow`](http://www.tensorflow.org/)
* [`tf_keras`](https://github.com/keras-team/tf-keras)
* [`pysoundfile`](https://pysoundfile.readthedocs.io/)

These are all easily installable via, e.g., `pip install numpy` (as in the
example command sequence below). Any reasonably recent version of these
packages should work.
packages should work. Note that YAMNet currently relies on Keras 2 and is
incompatible with Keras 3 (the default as of TF 2.16).

YAMNet also requires downloading the following data file:

Expand Down
2 changes: 1 addition & 1 deletion research/audioset/yamnet/yamnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@

import numpy as np
import tensorflow as tf
from tensorflow.keras import Model, layers
from tf_keras import Model, layers

import features as features_lib

Expand Down
Loading