search

axinc-ai / ailia-models

The collection of pre-trained, state-of-the-art AI models for ailia SDK
2.04k stars 325 forks source link

Implement Rnnoise #1361

Closed ooe1123 closed 8 months ago

ooe1123 commented 10 months ago

https://github.com/axinc-ai/ailia-models/issues/1224 のPRです。

kyakuno commented 9 months ago

モデルをアップロードしました。 https://storage.googleapis.com/ailia-models/rnnoise/rnn_model.onnx

kyakuno commented 9 months ago

実行すると下記のエラー。

Traceback (most recent call last):
  File "/Users/kyakuno/Desktop/repos/ailia-models-ax/audio_processing/rnnoise/rnnoise.py", line 521, in <module>
    main()
  File "/Users/kyakuno/Desktop/repos/ailia-models-ax/audio_processing/rnnoise/rnnoise.py", line 510, in main
    net = ailia.Net(MODEL_PATH, WEIGHT_PATH, env_id=env_id)
          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/usr/local/lib/python3.11/site-packages/ailia/wrapper.py", line 251, in __init__
    self._open_weight(weight)
  File "/usr/local/lib/python3.11/site-packages/ailia/wrapper.py", line 282, in _open_weight
    core.check_error(code, self.__net)
  File "/usr/local/lib/python3.11/site-packages/ailia/core.py", line 911, in check_error
    raise AiliaException(detail)
ailia.core.AiliaException: code: -128 (Unknown error.)
+ error detail : Blob(const_fold_opt__1293): Shape is mismatch at Blob::setTensor(). expected_shape is (1) (stride:(1)), but actual_shape is (2) (stride:(1))
kyakuno commented 8 months ago

python3 rnnoise.py --onnxだと動作する。

kyakuno commented 8 months ago

ailiaチームに確認中。

kyakuno commented 8 months ago

prototxtを更新して動作を確認しました。