Core MLのモデルの出力がMLMultiArrayの場合のメモ

2019年7月9日 16:24

画像分類みたいに出力が単純なクラスのインデックスではなく、画像を出力するようなCore MLモデルではよく出力の型がMLMultiArrayであることが多い。

たとえば公式のデプス推定モデル"FCRN-DepthPrediction"のOutputsのTypeはこんな感じ。

1 * 128 * 160の要素がDoubleなMultiArrayである、と。

単眼カメラで、Core MLのFCRN-DepthPredictionモデルを使ってデプス推定したもの。かなり良いのでは。ちなみに新APIは使ってないのでiOS 11+で動く。 pic.twitter.com/nDSVdLnPue
— Shuichi Tsutsumi (@shu223) July 9, 2019

このMultiArrayの扱いに関して、諸々のメモ。

MLMultiArrayのshapeのどの次元が幅で、高さで、チャンネルなのか

ここから先は

3,641字

¥ 300

ログイン

最後まで読んでいただきありがとうございます！もし参考になる部分があれば、スキを押していただけると励みになります。 Twitterもフォローしていただけたら嬉しいです。 https://twitter.com/shu223/