WindowsでAIプログラミング

TecoGAN:Windows向けコードをgithubで公開しました。

はじめに

『TecoGAN』は、Linuxベースのコードになっていますが、
要望があり、Windowsで実行できるように改変したものを作ったので、
GitHubで公開しました。
『WSL』を使いません。

https://github.com/supplepentan/tecogan-win

設定方法

前準備

CUDA(GPUが、必要です)

『CUDA 10.0』と『CUDA 10.1』を設定にしてください。
『Tensorflow-gpu==1.14』は、『CUDA 10.0』、
『Pytorch-cuda 』は、『CUDA 10.1』を使用するようになっています。
『Pytorch-cuda』は、超解像結果の評価コマンドのみに使用するので、
評価が必要ない方は、『CUDA 10.1』、『Pytorch-cuda』を設定しなくても問題ありません。

【参照】
CUDA:Windows10にCUDA10.0を追加設定

pyenv

pyenvで、『3.6.8』をローカル設定してください。
『tensorflow==1.14』を使用するため『3.6.8』に設定します。

【参照】
pyenv:Pythonのバージョン管理

実装

『PowerShell』を使います。

環境設定

GitHubからクローンを作成
カレントディレクトリを『tecogan-win』へ移動
仮想環境を作成し、アクティベイト
必要モジュールの設定

TecoGANコマンド

モデルとデータのダウンロード

サブプロセスに対応させていないので、結構時間がかかります。
いずれ対応させようと思います。

超解像の実行

実行結果は、オリジナルの『TecoGAN』と同じになります。
下記のページをご参照ください。
TecoGAN:Windows10に実装

結果の評価

さいごに

現段階では、『学習モード』が使用できないので、
今後は、これを使える様にしていこうと思います。(‘ω’)ノ

関連記事

TecoGAN:Windows10に実装
TecoGAN:Windows10に実装”nvidia-tensorflowを使って実装”

この記事を書いた人
化学系で博士号を取得したが、 あるとき、これからの時代はプログラミング!、と目覚める。 pythonを用いてデータ解析や機械学習に没頭。 最近は、フロントにVue.js、バックにFastAPIを用いて、 PytorchやTensorflowを実装したWebサービスを開発中。☆('ω')☆
SNSでフォローする
コメント (4)
  1. nori より:

    さぷりぺんたんさん初めまして。

    ノートパソコンでGPUがついていないWindows10で実装を考えています。

    TecoGAN:Windows10に実装で、全てのコードを処理しましたが、カレンダーのテスト処理をしたところ、
       Unknown command line flag ‘use_gpu’ .Did you mean:use_cprofile_for_profiling?

    というエラーが出ます。

    これは、GPUが搭載されていないノートパソコンでは実装できないという意味でしょうか?

    お忙しいと思いますが、お時間の許す時に回答を頂けたらと思います。

    1. さぷりぺんたん より:

      書き込みありがとうございますm(__)m
      と、返答がたいへん遅くなり申し訳ありませんm(__)m
      TecoGanですが、おいらがアップしているコードは、基本的にGPUありきのコードになっております。
      学習済みモデルを変換すればGPUがなくても実装可能だと思われまふ。
      が、TecoGan自体は、2グレードくらい前の理論になりまふ。
      目的に応じては、他のコードを参照した方が良いのでは、と思われまふm(__)m

  2. こっすー より:

    コメント失礼します。大学の研究でこのTeCoGANを使おうと考えているのですが、もとのコードの複雑さ、難しさ上、訓練データを変更したり、できた動画の定量的評価をすることは難しいでしょうか?

    1. さぷりぺんたん より:

      書き込みありがとうございまふm(__)m
      と、返答がたいへん遅くなり申し訳ありませんm(__)m
      TecoGanでふが、再学習することは可能でふが、目的によっては、他のコードを用いた方が良いと思われまふ。
      定量評価でふが、対照を固定すれば可能でふ。
      が、対照が目的に適していなければ、実用的には、あまり意味がないのでふm(__)m

コメントを残す

メールアドレスが公開されることはありません。

CAPTCHA