【Unity6.0+AI】手書き数字のSentis読み込みモデル認識実装事例

国際的な慣例に従って、その効果を見てみましょう。

 

材料の準備:

PSで黒背景に白文字の手書きフォントを描いて、サイズ28*28のjpgで書き出しました！

マテリアルの設定

基本的な手順

1. 準備: ONNX Model Zoo から手書き認識 ONNX モデル ファイルをダウンロードします。

2.  [モデル] MNIST 手書き数字認識モデル mnist-12.onnx をダウンロードし、プロジェクト ウィンドウの Assets フォルダーにドラッグします。

3.  [モデルのダウンロード] このケースではサンプル モデルが提供されますhttps://download.csdn.net/download/leoysq/88661406 https://download.csdn.net/download/leoysq/88661406

4. 【台本作成】シーンに台本を添付します。

5. モデルアセットを「ゲームオブジェクトインスペクター」ウィンドウの「modelAsset」フィールドにドラッグします。
6. 以下の digit.png 画像をダウンロードし、プロジェクト ウィンドウの Assets フォルダーにドラッグします。インポート設定で「2のべき乗以外」を「なし」に設定し、「適用」をクリックします。
7. digit.png をゲーム オブジェクト インスペクター ウィンドウの inputTexture フィールドにドラッグします。
8. 「再生」をクリックします。ゲームオブジェクトのインスペクター ウィンドウでは、結果配列の各項目が、モデルが画像を数値として予測する精度を示します。たとえば、配列の項目 0 は、モデルが画像が手書きのゼロであることをどの程度予測しているかを表します。

作者自身のコード:

コードのプログラム ロジック:

基本的なプロセスは次のとおりです。

1. ダウンロードモデル
2. Unity アセット フォルダーに配置します (または独自のフォルダーを作成します)。
3. Sentis プラグイン パッケージに従ってください。
4. Unity.Sentis を使用してスクリプト内で参照されます。
5. アセットをコードに取り込む
6. このアセットをモデルとしてロードします
7. モデルに簡単に入力できるようにテンソル データを作成する
8. ワークフロー スレッドを作成します。 IWorker 変数 = WorkerFactory.CreateWorker();
9. ワークフロー操作の実行テンソル データ oneWorkLineAI.Execute(InputTensor);
10. 出力データ (デフォルトのテンソル形式と