Resnetの転移学習について

Question

オリジナルの学習データを用いて、Resnetで転移学習をしようとしています。

Resnet18を用いた転移学習をしているサイトなどご存知の方いましたら、教えていただきたいです。
私が見たところ、VGGでの転移学習をしている方は多いのですが、Resnetで転移学習をしている方を見つけられていません。

よろしくお願いします。

Answer 1

Google検索 - 「resnet18 transfer learning」
Google検索 - 「resnet18 fine tuning」

---

Pytorch tutorial - Finetuning the convnet
github - pytorch/tutorials/beginner_source/transfer_learning_tutorial.py
人工知能に関する断創録 - PyTorch (8) Transfer Learning (Ants and Bees)

Comments

[syayo]

ありがとうございます。

Answer 2

pytorch関連

https://discuss.pytorch.org/t/how-to-perform-finetuning-in-pytorch/419/21

https://www.kaggle.com/pytorch/resnet18/data

---

torchからtensorflow。
ちょっとバージョンが古い。

https://github.com/dalgu90/resnet-18-tensorflow/blob/master/README.md

https://gist.github.com/farrajota/07cf6d926ecc028d5135

Comments

[syayo]

ありがとうございます。
Resnetを扱う際はPyTorchで行なうのが普通なのでしょうか？
私はPyTorchに触れたことがないので、少し抵抗を感じています。

[mkgrei]

転移学習ということで、訓練済みの重みが必要になります。
なぜtorchの重みが人気であるのかはわかりませんが、
https://github.com/facebook/fb.resnet.torch/tree/master/pretrained
facebookが権威的に重みを配布しているためではないでしょうか。

https://github.com/dalgu90/resnet-18-tensorflow/blob/master/README.md
こちらのリンクでは、この重みをtensorflow用に変換してから使用しています。

ただの重みなので、対応付けることができれば、フレームワークには依存しません。

[syayo]

ありがとうございます。
今後試してみようと思います。