線画をイラストに変換するAIを動かしてみる

線画をイラストに変換するAIを動かしてみた。機械学習には時間がかかったが、結果として、線を描き足す都度、イラストが変化し更新されるツールを楽しめた。

結果

線画（左）を描けば、イラスト（右）が出力される。線を描き足す都度、イラストが変化し更新される。その様子を動画にした。

コードと使い方は下記の書籍の Chapter 07 を参照した。

コードは、ほとんど編集せずそのまま使った。バッチサイズだけは、BATCH_SIZE = 64 と書き換えた。

アニメの顔のイラストを線画にして、元画像とセットでAIに渡し、機械学習で線画からイラストを生成するモデルを作り、そのモデルを動かす流れになる。

機械学習でモデルを作成するためにGPUを動かしている時間が、6時間ほどかかった。すぐに結果を見れるわけでなく、夜間に動かして翌朝に確認した。BATCH_SIZEはまだまだ大きくできる気がした。

PyTorchの勉強をして、新しいグラボを存分に活用したいと考えた。