ホーム > おもちゃのAI研究室 >

顔の特徴操作・顔の合成(StarGAN)

人工知能で「顔の特徴」(属性)の操作、「二人の顔を合成」します。

※属性は人物の若さ(年齢)、性別、髪の色、ヒゲなど40種類を変更できます。

左から順に元画像、黒髪、金髪、茶髪、性別(男性)、年齢(若く)です。

「StarGAN-v1」はこのような結果です。「StarGAN-v2」だと精度はかなり向上しています。

このAI(人工知能)の結果は調査及び研究・教育目的で公開しています。

本来は改良した「TensorFlow Lite」を用いて学習済みモデルで推論する予定でしたが、結果のみを公開しています。

StarGAN-v2の精度は高いのですが非商用に限るなので、ライセンス関係で検討するとGlowの方が良いかも知れませんね。ただ、CelebAのようなデータセットは自前で用意する必要があります。※日本の法律だとCelebAも商用で使えるかも知れません。

次に備忘録としてStarGANの使い方を書いておきますね。

StarGAN(v1)の使い方

StarGAN-Tensorflow\dataset\celebA\testに変換対象となる画像データを配置します。後はcheckpointを指定して実行します。

python main.py ^
--phase=test ^
--checkpoint_dir=C:/Users/xxx/Desktop/checkpoint

CelebAのファイルが存在しないエラーをなくしたい場合はStarGAN.pyの428行目から447行目をコメントにして下さい。

※StarGAN-v2はColab ProでPyTorch版を実行しました。公式を参照すればわかります。