Mune's Weblog

鹿児島在住のstray radiologistです
仕事も趣味も画像関係…

2021.01.11 Windows 10でCycle GAN(Pytorch実装)

2021年01月11日 | 仕事関係
なんとなくGAN(敵対的生成ネットワーク)を避けて通っていたのですが、ちょっとしたアイデアがあり色々調べているうちに、Cycle GANが使えそうな気がしてきました。

原著はコチラ
Unpaired Image-to-Image Translation using Cycle-Consistent Adversarial Networks
参考にしたQiitaはコチラ
CycleGANをpytorchで実装してみた

CycleGANの再現実装してみた

実装は前のQiitaの著者が作られたGit hub(Linux用)をWindows10のコマンドラインから実行。
画像は2番目の著者のスクリプトを読んで見つけた本家のサイトから、horse2zebra.zipをDLして使用。
TITAN RTXでほぼ丸一日かけて200エポックの学習。



向かって左から original, fake, after cycleの順。
ひとまず確認できたので、次はオリジナルデータを作って仮説の検証です。

最新の画像もっと見る

コメントを投稿