美少女イラストのタグ付けを行う「Deep Danbooru」のソースコードをgithubに公開しました。
リポジトリ URL:
https://github.com/KichangKim/DeepDanbooru
前回のバージョンとの違いは以下の通りです：
- MLライブラリをCNTKからTensorFlow 2に変更
- OptimizerをAdamからSGD with momentumに変更
- Danbooruデータセットを20191108に更新
CNTKとTensorFlowの中の処理がだいぶ違うので全く同じネットワークとデータセットでも結果がかなり違います。今回のリリースではできるだけCNTKの結果に近づけることを意識しました。
OptimizerをAdamからSGDに変えることで一般化性能が多少向上した気がしますが、未検証です。
