ADD semantic-aware-low-light-image-enhancement

axinc-ai / ailia-models

The collection of pre-trained, state-of-the-art AI models for ailia SDK

1.91k stars 311 forks source link

Open kyakuno opened 11 months ago

kyakuno commented 11 months ago

Akasan commented 8 months ago

モデルがforward1とforward2の二段階で実行されるようになっているため、一つのforwardにまとめる必要あり
元コードは結構修正をしないとサンプルなども動かない部分がある（イシューを見ればコメント残っていて対応可能）
forwardがに段階になっていたりして、ONNX変換はできるが利用しづらい部分がある
- 一つのforwardで動かせるように実装を調整 →ONNX一つに合わせることで動くことを確認（ただし、特定のオプションだけなので、別のオプションのモデルはまだ）

Akasan commented 7 months ago

opset_versionの問題で、コードの変更をせずにonnxに完全に変換するのは難しそう
- コードがかなり煩雑なので、実際に対応するとなるとかなりの工数がかかるのではと予測します

Akasan commented 6 months ago

エクスポートするときにsymbolic_helper周りでエラーが発生する

TypeError: 'NoneType' object is not subscriptable
(Occurred when translating col2im).

Akasan commented 6 months ago

@kyakuno 対象レポジトリのうち、DRBN_SKFはONNXへの変換が成功しましたが、LLFlowとSNR_SKFは変換時点でエラーが発生している状況です。コード全体を見直して動かせるようにできる可能性はありますが、どれくらいかかるかどうかが不透明な状態です。

一旦DRBN_SKFはv1とv2ともにONNX変換できているのでailai-modelsに実装をしてみようと思うのですがいかがでしょうか？

kyakuno commented 6 months ago

@Akasan はい、まずはDRBN_SKFだけの登録でも問題ありません。

Akasan commented 6 months ago

@kyakuno DRBN_SKFだけのものはもう少しで実装が完了しますので、完了次第PR作成します