ヨメミか萌実か判別するAIのモデル生成に使用されたソースコードです
※GPU使わない場合は tensorflow-gpuを tensorflowに変えてインストールしてください
・Windows10 2004 64bit
・Anaconda 1.19.2 (Python3.8.5)
・AMD Ryzen5 3600
・RAM 32GB
・NVIDIA GeForce RTX 2060 SUPER
requirements.txtちょっと適当に作ってしまったので間違ってたらPRでも送っていただけると幸いです
pip3 install -r requirements.txt
まずMoemi
、Yomemi
フォルダにそれぞれ画像を集めてきます
次に
python3 img.py
を実行して画像ファイルをnumpyのバイナリに変換します
そうしたらdataset.npy
が生成されます
python3 train.py
これでモデルのトレーニングが開始されます
データセットの数や環境に応じてソース上のepochs=20
の数値を変えてください
適切な値にしないと学習不足、過学習になる場合があります
トレーニングが終わるとcnn.h5
というkerasのモデルが生成されます
また、loss
とaccuracy
の変化を表したグラフも生成され表示されます
python3 think.py <画像ファイルへのパス>
引数に画像ファイルのパスを指定します
tensorflowが動き始めて確率と結果が表示されれば成功です