pytorchで画像認識のモデルを作って。テストデータを評価する、って記事はたくさんありましたが、JPEGとかPNGとかの画像を実際に読み込んで予測してみた記事があんまりない気がしたので、まとめてみました。 今回のゴール PNG画像をPyTorchで作った学習モデルに通して予測してみます。 モデルはMNISTの手書き数字認識を使います。 学習モデルの構築 Google ColaboratoryでPyTorchでMNISTを学習したモデルを保存し、それを読み出して使う簡単サンプル - 人工知能プログラミングやってくブログ この記事を参考に学習モデルを作ります。 動かすと1,725,616バイトのmnist_cnn.ptができました。 機械学習モデルを使って予測する PyTorch 1.1 Tutorials : 画像 : PyTorch を使用した画風変換 – PyTorch この記事を参考