「Deep Danbooru」のソースコード公開

美少女イラストのタグ付けを行う「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に変えることで一般化性能が多少向上した気がしますが、未検証です。