はじめに
Zigで機械学習やりたいなと考えていたところ、@mattn様がまた面白いものを作ってくださったので、早速遊んでみた。
https://zenn.dev/mattn/articles/af64c6a3eefad0
とりあえずMNIST
というわけでMNISTをやってみることにした。 レポジトリはこちらにある。
https://github.com/ryoppippi/zig-tflite-mnist
READMEに従って、上のレポジトリをCloneし、TensorFlow Liteのインストールを済ませ、kaggleからDatasetを落としてきたらあとはビルドするのみ！
一応今回推論に使用するモデルもCommitしてあるが、自分で学習することももちろんできる。 学習に用いたNotebookはこちら↓
画像の読み込みについて
mattn氏の記事でも触れられていた通り、Zigで画像を扱うためのライブラリが本当に少ない。 使えるもののうち、開発が進んでいるものも発展途上だったりする。
https://github.com/zigimg/zigimg
しかし、C言語とZigの相性が抜群なことは皆様ご存知だろう。 今回はヘッダーオンリーで画像の読み込みやリサイズが出来て、かつパブリックドメインで使えるSTBライブラリを使用した。
https://github.com/nothings/stb
Zigから本当に簡単に呼び出せたので、Zig使いで画像の扱いに悩んでいる方は是非とも試してみてほしい。
終わりに
ZigからTensorflow Liteを呼び出せるライブラリを使って、MNISTをやってみた。 画像を扱う機械学習もある程度ならZigでできることがわかった。 皆さんも楽しんでみてください。