二値分類モデル

Deep Learning の実務では、画像分類モデルの学習が主要なタスクの一つです。2つのクラスを見分ける二値分類に出会うことも多いでしょう。ワークフローを効率化し再利用性を高めるため、今後のプロジェクトにも適用できる二値画像分類用の CNN モデルのテンプレートを作成することにしました。

torch と、torch.nn を nn としてインポート済みです。すべての画像サイズは 64x64 ピクセルです。