More than 3 years have passed since last update.

インド人になる(Cycle-GANによるドメイン変換)

Posted at 2021-01-09

はじめ
インドで本場のカレーを堪能したいのですが、コロナ渦で中々それが叶わず。なので現地に行けないのなら、せめてインディアンスの顔がインド風になればそのカレー欲が満たされるのでは(？)と思いその仮説を検証することにしました。

なかみ
CycleGANでインド人の顔と日本人の顔のドメイン変換器を作ります。
以下を参考に実装しました。
CycleGANをpytorchで実装してみた

学習に必要なインド人の顔の画像と日本人の顔の画像は以下を参考に取得しました。
[API を叩かずに Google から画像収集をする]
(https://qiita.com/naz_/items/efc296ae1bf0e62f6704)

image_collector.py

class Google(object):
    def __init__(self):
        self.GOOGLE_SEARCH_URL = "https://www.google.co.jp/search"
        self.session = requests.session()
        self.session.headers.update(
            {
                "User-Agent": "Mozilla/5.0 (X11; Linux x86_64; rv:10.0) \
                    Gecko/20100101 Firefox/10.0"
            }
        )

のself.GOOGLE_SEARCH_URLは"http://www.google.co.jp/search"
に変えました。SSLの認証がうまく通らなかったようです。

蛇足ですが”日本人男”などのワードで検索しても外人女性などの写真が多く紛れていて除去作業が大変でした。”海外女性と結婚する日本人男性の特徴5選！”のような記事の画像を入手していたっぽいです。
"indian men"とか"indian male"だとインド人男性と思われる顔を多く入手できるのですが。

顔の切り出しについてはopenCVのカスケード分類でやりました。
[PyTorchを使って日向坂46の顔分類をしよう！]
(https://qiita.com/coper/items/b1fd51062642d624e26f)

こんな感じの画像(128*128)がそれぞれ400枚ほど集まりました。