implement crestero

[yuki399]

#796

[kyakuno]

ailia SDK 1.2.11では下記のエラー。

ailia.core.AiliaInvalidLayerException: code: -10 (Incorrect node parameter. [broken or unsupported AI model file])
+ error detail : Layer:onnx::ReduceSum_2526(Eltwise) Error:Unsupported size pair. [virtual shape] 

[kyakuno]

ailia SDK 1.2.12 (alpha)でも同じエラー。

[kyakuno]

ailia SDK 1.2.13で対応の方針。

[kyakuno]

ailia SDK 1.2.13ではモデルは推論できていて、後処理がうまく動いていない。

  File "crestereo.py", line 187, in <module>
    main()
  File "crestereo.py", line 184, in main
    recognize_from_image(net)
  File "crestereo.py", line 113, in recognize_from_image
    disparity_map = depth_estimator(left_img, right_img)
  File "/Users/kyakuno/Desktop/repos/ailia-models/depth_estimation/crestereo/crestereo_util.py", line 22, in __call__
    return self.update(left_img, right_img)
  File "/Users/kyakuno/Desktop/repos/ailia-models/depth_estimation/crestereo/crestereo_util.py", line 62, in update
    self.disparity_map = self.process_output(outputs)
  File "/Users/kyakuno/Desktop/repos/ailia-models/depth_estimation/crestereo/crestereo_util.py", line 99, in process_output
    return np.squeeze(output[:,0,:,:])
TypeError: list indices must be integers or slices, not tuple

[kyakuno]

predictの後でonnx runtimeと同様に[0]としたら動いた。