PyTorchで画像分類(その2)

前回は画像ファイルを読み込むためのDatasetを作成しましたが、今回はディープラーニングの魂であるモデルの作成をします。 題材は前回と同じkaggleの犬/猫の画像分類コンペを使います。 モデル(Model)の実装 ではモデルの実装をしていきます。 とは言っても、モデルとなるネットワークを一から実装することはせずに既存のネットワークを拝借します。 今回はR […]

PyTorchで画像分類(その1)

PyTorchで画像分類をやってみたので、何回かに分けて解説します。 題材としては画像分類のシンプルな問題であるkaggleの犬/猫の画像分類コンペを使います。 データセット(Dataset)の実装 このコンペでは画像ファイルが提供されますが、PyTorchで処理するためにはテンソル形式に変換する必要があります。また、ディープラーニングでは読み込んだ画像を単 […]