ADD: Anytime Stereo Image Depth Estimation on Mobile Devices about ailia-models HOT 3 OPEN

サンプルコード(finetune.py)を動作させるために、Scene Flow(https://lmb.informatik.uni-freiburg.de/resources/datasets/SceneFlowDatasets.en.html) のダウンロードは必要なし。
KITTI Stereo(http://www.cvlibs.net/datasets/kitti/eval_stereo.php) の2015データ(data_scene_flow.zip)のみをダウンロードする。

data_scene_flow.zip を解凍し、作成される trainingディレクトリ配下のimage_2とimage_3ディレクトリ配下のファイルがそれぞれモデルの入力のimgLとimgRになる。
正解データはdisp_occ_0ディレクトリ配下のファイルを想定。

[推論実行]
python finetune.py --maxdisp 192 --with_spn --datapath ./training/ --save_path results/kitti2015 --datatype 2015 --pretrained checkpoint/kitti2015_ck/checkpoint.tar --split_file checkpoint/kitti2015_ck/split.txt --evaluate

from ailia-models.

onnxエクスポートで以下のエラーが発生。

RuntimeError: Exporting the operator grid_sampler to ONNX opset version 11 is not supported.

grid_sampler がonnxでサポートされていないため、エクスポート不可能。

from ailia-models.

spnモジュールからインポートしているGateRecurrent2dnoindもエクスポートできない

RuntimeError: ONNX export failed: Couldn't export Python operator GateRecurrent2dnoindFunction

from ailia-models.