この記事の読者
深層学習・ディープラーニングの手法の1つである「EfficientNet V2」について知りたい.
Index
EfficientNet V2 とは
#アルゴリズム編 yhayato1320.hatenablog.com
Dataset と Task
Dataset は、MNIST を利用
タスク は 画像認識 (10 分類問題)
環境とライブラリ
- PyTorch (1.10.0)
- timm (PyTorch Image Model) を利用.
実装
EfficientNet #アルゴリズム編をほぼほぼ利用. (モデルの実装の部分のみ変更)
データの取得
モデル定義
EfficientNet 系 (MobileNet 系) はここから選べる様子.
今回は、ここから efficientnetv2_s を選択.
Train
Infer
参考
- 【PyTorch×転移学習】学習済みモデルライブラリTIMMのご紹介
- timm ライブラリの使い方
- dajiro.com