PyTorch 1.1 Tutorials : 画像 : ONNX を使用して PyTorch から Caffe2 とモバイルにモデルを移す (翻訳/解説)
【PyTorch 1.1 Tutorials: 超解像 - ONNX による Caffe2 とモバイルへの移行】
PyTorch 1.1 リリースのチュートリアルの再翻訳を進めています。
今回は「画像」カテゴリーから ONNX を使用して PyTorch から Caffe2、更にモバイル環境にモデルを移行するチュートリアルです。
このチュートリアルでは、超解像ネットワーク (Super-Resolution) を題材にして、PyTorch で定義された超解像モデルをどのように ONNX フォーマットに変換するか、そしてどのように Caffe2 にロードするかについて説明します。更にモバイルデバイス (Android) へのロード・実行方法も解説します。
PyTorch は TensorFlow とともに多く利用されている深層学習フレームワークです。5 月に PyTorch 1.1 がリリースされ 1.0 のドキュメントから追加や修正が入っていますので、順次再翻訳しています。