• Manoj Plakal's avatar
    Added TF-Lite-compatible feature extractor and model exporter for YAMNet (#9098) · 8da48573
    Manoj Plakal authored
    * Added TF-Lite-compatible feature extractor and model exporter for YAMNet.
    
    - Added a TF-Lite compatible feature extractor. With the latest TF-Lite,
      that involves a DFT-multiplication replacement for tf.abs(tf.signal.stft())
      and not a lot else. Note that TF-Lite now allows variable-length inputs.
    - Added a YAMNet exporter that produces TF2 SavedModels, TF-Lite models,
      and TF-JS models.
    - Cleanups: switched hyperparameters to a dataclass, got rid of
      some lingering cruft in yamnet_test.
    
    * Responded to DAn's comments in https://github.com/tensorflow/models/pull/9098
    
    - Switched some hparams to float
    - Made class map asset available on the exported model, and tested that
      it can be loaded from the various exports.
    8da48573
features.py 7.31 KB