tensorflow / models

Models and examples built with TensorFlow
Other
77.15k stars 45.77k forks source link

Using YAMNet in android #8475

Closed paleomoon closed 4 years ago

paleomoon commented 4 years ago

System information

Please provide the entire URL of the model you are using?

https://github.com/tensorflow/models/tree/master/research/audioset/yamnet Describe the current behavior I'm using YAMNet in android, I compiled Tensorflow 1.13 from source in Ubuntu 16.04 for android platform, so I can use native c++ to do inference in android. But now I get no output of YAMNet. This is the logcat in android studio, seems the 'RFFT' op is not supported?

D/Mudra: Graph /data/user/0/com.example.sd/cache/yamnet.pb read successfully! 
I/Mudra: Create TensorFlow Session: Invalid argument: No OpKernel was registered to support Op 'RFFT' used by {{node log_mel_features/stft/rfft}}with these attrs: []
    Registered devices: [CPU]
    Registered kernels:
      <no registered kernels>

         [[{{node log_mel_features/stft/rfft}}]]

Describe the expected behavior I test the pb model in windows, tensorflow 1.14, it works fine. Code to reproduce the issue

N/A Other info / logs

N/A

paleomoon commented 4 years ago

It seems to be a tf version problem, I will try to compile tf 1.14 again, though failed before.