2022年1月5日
WindowsでAIプログラミング
Lightweight-GAN：Windows10に実装
目次
はじめに
すごい簡単にインストールできました。
pipのみでインストールできるGAN、なんと簡単(‘ω’)
参照サイト
https://github.com/lucidrains/lightweight-gan
PC環境
Windows10 pro (Windows Insier Program)
マウスコンピューター製G-Tune E5-144
CPU：インテル(R) Core(TM) i7-10875H プロセッサー ( 8コア / 16スレッド / 2.30GHz [ 最大5.10GHz ] / 16MBキャッシュ )
メモリ：32GB メモリ [ 16GB×2 ( PC4-21300 / DDR4-2666 ) / デュアルチャネル ]
SSD (M.2)：512GB NVMe SSD ( SAMSUNG PM981a / M.2 PCIe Gen3 x4 接続 )
グラフィックス：NVIDIA GeForce RTX2060 / 6GB
実装方法
仮想環境の構築
|
1 |
python -m venv venv |
仮想環境をアクティヴェイト
|
1 |
vevn/scripts/activate |
Pytorch（CUDA10.1対応）のインストール
|
1 |
pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html |
LightWeight-GANのインストール
|
1 |
pip install lightweight-gan |
実行方法
実行コマンド
|
1 |
lightweight_gan --data ./data --name [name] --batch-size 16 --gradient-accumulate-every 4 --num-train-steps 100000 --aug-types [translation,color] |
[name]は任意、フォルダが作成されます。
学習済みモデルからGIFを作成
|
1 |
lightweight_gan --name [name] --generate-interpolation |
化学系で博士号を取得したが、
あるとき、これからの時代はプログラミング！、と目覚める。
pythonを用いてデータ解析や機械学習に没頭。
最近は、Pytorchで作ったONNXモデルを、Nuxt3にのせたWebサービスの開発、
ChatGPT や Stable Diffusion に没頭中☆('ω')☆
SNSでフォローする
関連記事
