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 Audiosep #1332

Open kobakos opened 11 months ago

kobakos commented 11 months ago

https://github.com/axinc-ai/ailia-models/issues/1275

kyakuno commented 9 months ago

モデルを下記にアップロードしました。 https://storage.googleapis.com/ailia-models/audiosep/audiosep_resunet.onnx.prototxt

kyakuno commented 9 months ago

ailia SDK 1.2.15と16だとSEGVになる。CPUモードでも同じ問題が発生する。

Thread 0 Crashed::  Dispatch queue: com.apple.main-thread
0   libailia.dylib                         0x12f636834 ailia::Layer::PadLayer::padConst(ailia::Tensor&, ailia::Tensor const&, ailia::TensorUtil::Shape const&, ailia::TensorUtil::Shape const&, float)::$_2::operator()(unsigned int, unsigned int) const + 3796
1   libailia.dylib                         0x12f6329b0 ailia::Layer::PadLayer::padConst(ailia::Tensor&, ailia::Tensor const&, ailia::TensorUtil::Shape const&, ailia::TensorUtil::Shape const&, float) + 864
2   libailia.dylib                         0x12f634130 ailia::Layer::PadLayer::_computeCpu() + 2192
3   libailia.dylib                         0x12f51241f ailia::Layer::DNNLayerBase::_compute() + 1295
4   libailia.dylib                         0x12f50d250 ailia::Layer::LayerBase::compute() + 112
5   libailia.dylib                         0x12f4e5db8 ailia::Layer::Graph::inferCore(AILIAGraphicsInteropDesc*, unsigned int) + 1128
kyakuno commented 9 months ago

@kobakos こちら、実行に成功した環境を教えていただければと思います。

kobakos commented 9 months ago

すみませんこれ送るファイル(audiosep_resunet.onnx)を間違えてしまっていたみたいです。 googleストレージの方ではsegmentation faultが出たのですが自前の環境にある正しいファイルを使うとCPUでもGPUでも(どちらもAMD)サンプルは実行できました。ただそちらの方でも入力のサイズを大きくするとsegmentation faultが出てしまうようです。