search

axinc-ai / ailia-models

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

Implement Marigold #1379

Closed ooe1123 closed 5 months ago

ooe1123 commented 8 months ago

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

kyakuno commented 8 months ago

PBが3.23GBと巨大なモデル。

kyakuno commented 8 months ago

モデルアップロード済み。 https://storage.googleapis.com/ailia-models/marigold/text_encoder.onnx

kyakuno commented 6 months ago

diffusionを使用した奥行き推定モデル。

kyakuno commented 6 months ago

スクリーンショット 2024-03-04 10 13 54 https://marigoldmonodepth.github.io/

kyakuno commented 6 months ago

@ooe1123 onnxruntimeで実行すると、下記のエラーになるようでして、text_encoder_weights.pb（Google Driveだと1.36GB）が破損している可能性がある気がしていまして、ご確認いただいても良いでしょうか？

onnxruntime.capi.onnxruntime_pybind11_state.InvalidProtobuf: [ONNXRuntimeError] : 7 : INVALID_PROTOBUF : Load model from text_encoder_weights.pb failed:Protobuf parsing failed.

ailiaだと下記でした。

ailia.core.AiliaBrokenDataException: code: -4 (The specified file was broken.)
+ error detail : Unexpected protobuf data type(6).
kyakuno commented 6 months ago

ファイルは下記のものを使用しています。また、ファイルの上書き権限を付与させていただきました。 https://drive.google.com/drive/u/2/folders/1PDGBp0I_Cod1L4-QUKisKkitzaFstUNl

kyakuno commented 5 months ago

@ooe1123 動作確認できました！ご対応、ありがとうございました。