Web上でディープラーニングを行うなんて、といった意見は過去によく聞かれました。しかしすでにネットワークを組んで訓練も終わったモデルを使うのであればJavaScriptからでも有益に使えます。エッジでディープラーニングの処理ができるならばユーザにとってもメリットがあります。 そんな可能性を飛躍的に拡大してくれるのがONNX.jsです。Webブラウザ上で使えるONNXモジュールです。

ONNX.jsの使い方

写真からオブジェクトを判定するデモ。

フォークリフト、だと…。

手書き認識。精度はかなり高いです。

感情分析。Webカメラを使ってみたのですが精度は低かったです…。

ONNX.jsはディープラーニングエンジンの共通モデルフォーマットとなっているONNXを使って動作します。つまりTensorflowやCaffe2、nnablaなど多くのディープラーニングライブラリで作成できます。Webブラウザ上でディープラーニングを実行したい時に便利なライブラリです。

ONNX.jsはTypeScript製のオープンソース・ソフトウェア（MIT License）です。

ONNX.js Microsoft/onnxjs: ONNX.js: run ONNX models using JavaScript