DeepDanbooruNSFWモデルを作成してみた

最新版は2022年くらいに更新が止まってしまっているので、新しいキャラクターなどはどうしても判別できません。なので新しいデータセットでモデルを学習させてみました。

GitHub - KichangKim/DeepDanbooru: AI based multi-label girl image classification system, implemented by using TensorFlow.

このリポジトリを参考に、windows 11環境下で、モデルをGPUを使用して学習させてみました。

簡単な手順としては 、

• Danbooru のユーザー登録、APIキーを取得。
• リポジトリのReadme通りのデータセット構造でデータをダウンロードするようにするスクリプトを作成
• Python 3.11: 推奨のバージョン
• TensorFlow 2.10.0
• CUDA Toolkit 11.2
• cuDNN 8.1: プロジェクト直下の cuda フォルダ内に配置し、実行時にパスを通す。