AIdemo の履歴(No.10) - PukiWiki

[ トップ ] [ 一覧 | 検索 | 履歴 | ログイン ]

履歴一覧
差分を表示
現在との差分を表示
ソースを表示
AIdemo へ行く。
- 1 (2024-08-27 (火) 13:55:56)
- 2 (2024-08-27 (火) 14:18:40)
- 3 (2024-08-28 (水) 17:10:25)
- 4 (2024-08-28 (水) 19:46:26)
- 5 (2024-09-01 (日) 23:04:09)
- 6 (2024-09-11 (水) 14:43:18)
- 7 (2024-09-21 (土) 14:13:28)
- 8 (2024-09-21 (土) 17:49:23)
- 9 (2024-10-12 (土) 20:22:04)
- 10 (2024-11-04 (月) 13:49:12)

私的AI研究会 > AIdemo

AI demo 起動プログラム†

　作成した各種「AI プログラム」を簡単に起動できるようにする
　※ 以下のページは新規ページ AI demo 起動プログラム (GUI 対応版) へ全面移行する

AI demo 起動プログラム
- AI demo「demo.py」
- 更新履歴
参考資料

※ 最終更新:2024/11/04　

AI demo「demo.py」†

ターミナル起動直後の環境から実行†

　Windows の場合　

(base) PS C:\anaconda_win> ./gan_test.ps1

　Linux の場合　
```
(base) ~$ source gan_test.sh
```

'entar' のみを入力すると終了する
終了後のディレクトリは「workspace_2」となる

仮想環境(py38_learn) 移行後に再度プログラムを起動する場合†

カレントディレクトリ「workspace_2/」
```
(py38_learn) python demo.py
```

「demo_cli.py」コマンド実行とプログラム選択†

コマンド

(py38_learn) python demo_cli.py

------------------------------
|  AI sample program (CPU)   |  Ver 0.02
------------------------------
# conda environments:
#
base                     C:\Users\izuts\anaconda3
py38_learn            *  C:\Users\izuts\anaconda3\envs\py38_learn
py38_learn_test          C:\Users\izuts\anaconda3\envs\py38_learn_test

Select input?   0 - 9 'enter'=終了
                0: fase_pasing          メークアップをシュミレーション
                1: DLFS                 人間の年齢による顔の変化
                2: encoder4editing      StyleGANを使った画像編集
                3: CycleGAN             画像スタイル変換
                4: stargan-v2           顔の合成
                5: QMUPD                似顔絵生成
                6: PAMA                 画像に別の画像のスタイルを転送(静止画)
                7: PAMA(movie)          画像に別の画像のスタイルを転送(動画)
                8: BlendGAN             顔画像に様々なスタイルをブレンドする
                9: StyleGAN3            StyleGAN による画像の編集
               10: StyleGAN3            StyleGAN + CLIP による画像の編集

------------------------------
|    program end..           |
------------------------------

入力 0: fase_pasing　メークアップをシュミレーション

入力 1: DLFS　人間の年齢による顔の変化

入力 2: encoder4editing　StyleGANを使った画像編集

入力 3: CycleGAN　画像スタイル変換

入力 4: stargan-v2　顔の合成

入力 5: QMUPD　似顔絵生成

入力 6: PAMA　画像に別の画像のスタイルを転送(静止画)

入力 7: PAMA(movie)　画像に別の画像のスタイルを転送(動画)

入力 8: BlendGAN　顔画像に様々なスタイルをブレンドする

入力 9: StyleGAN3　StyleGAN による画像の編集

入力 10: StyleGAN3　StyleGAN + CLIP による画像の編集

「demo_cli.py」詳細†

およその所要時間

入力	選択プログラム		GPU	CPU
0	fase_pasing	メークアップをシュミレーション	-	-
1	DLFS	人間の年齢による顔の変化	数十秒	- ※
2	encoder4editing	StyleGANを使った画像編集	数分	- ※
3	CycleGAN	画像スタイル変換	数十秒	数分
4	stargan-v2	顔の合成	数分	数十分 ※
5	QMUPD	似顔絵生成	数十秒	数分
6	PAMA	画像に別の画像のスタイルを転送(静止画)	数秒	数十秒
7	PAMA(movie)	画像に別の画像のスタイルを転送(動画)	数十秒	数十分 ※
8	BlendGAN	顔画像に様々なスタイルをブレンドする	数十秒	- ※
9	StyleGAN3	StyleGAN による画像の編集
10	StyleGAN3	StyleGAN + CLIP による画像の編集

　※ 処理結果がある場合は表示のみの時間で終了する
　・終了は 'enter' 入力

GPU/CPU 動作
・GPU動作環境を自動的に識別する
・GPU搭載機種で CPU動作をする場合は --cpu オプションを付加する
```
(py38_learn) python demo.py --cpu
```

モジュール・ソースコード

▼「demo_cli.py」

# -*- coding: utf-8 -*-
##------------------------------------------
##  AI demo link program       Ver 0.02
##
##               2024.08.25 Masahiro Izutsu
##------------------------------------------
## demo.py → demo_cui
##  Verr 0.02       2024.09.19  StyleGAN3 追加

import warnings
warnings.simplefilter('ignore')

# Color Escape Code
GREEN = '\033[1;32m'
RED = '\033[1;31m'
NOCOLOR = '\033[0m'
YELLOW = '\033[1;33m'
CYAN = '\033[1;36m'
BLUE = '\033[1;34m'

from torch.cuda import is_available
gpu_d = is_available()                                          # GPU 確認

# インポート＆初期設定
import os
import sys
import platform
from os.path import expanduser

# 定数定義
CONDA_ROOT = '/anaconda_win/' if platform.system()=='Windows' else '~/'
WORK_ROOT = expanduser(CONDA_ROOT + 'workspace_2')
# ** main関数 **
def main(gpu_d):
    os.system('conda info -e')
    os.chdir(WORK_ROOT)

    msg = "Select input?\t0 - 9 'enter'=終了\n\
                0: fase_pasing\t\tメークアップをシュミレーション\n\
                1: DLFS\t\t\t人間の年齢による顔の変化\n\
                2: encoder4editing\tStyleGANを使った画像編集\n\
                3: CycleGAN\t\t画像スタイル変換\n\
                4: stargan-v2\t\t顔の合成\n\
                5: QMUPD\t\t似顔絵生成\n\
                6: PAMA\t\t\t画像に別の画像のスタイルを転送(静止画)\n\
                7: PAMA(movie)\t\t画像に別の画像のスタイルを転送(動画)\n\
                8: BlendGAN\t\t顔画像に様々なスタイルをブレンドする\n\
                9: StyleGAN3\t\tStyleGAN による画像の編集\n\
               10: StyleGAN3\t\tStyleGAN + CLIP による画像の編集\n"

    loopf = True
    while loopf:
        val = input(msg)
        if val == '0':
            path = WORK_ROOT + '/motion-co-seg/face_parsing'
            prog = 'makeup_gui.py'

        elif val == '1':
            path = WORK_ROOT + '/DLFS'
            prog = 'dlfs_gui.py'

        elif val == '2':
            path = WORK_ROOT + '/encoder4editing'
            prog = 'e4e_gui.py'

        elif val == '3':
            path = WORK_ROOT + '/CycleGAN'
            prog = 'gan_paint.py --source_dir ./images'

        elif val == '4':
            path = WORK_ROOT + '/stargan-v2'
            prog = 'starv2_sample.py --src_dir ./assets/representative/celeba_hq/src_0 --ref_dir .//assets/representative/celeba_hq/ref2_00'

        elif val == '5':
            path = WORK_ROOT + '/QMUPD'
            prog = 'qmupd_demo.py --source_dir ./examples'

        elif val == '6':
            path = WORK_ROOT + '/PAMA'
            prog = 'pama.py --source_image ./content/10.jpg --style_image ./style/03.jpg'

        elif val == '7':
            path = WORK_ROOT + '/PAMA'
            prog = 'pama_movie.py --source_image ./video/02.mp4 --style_image ./style/03.jpg'

        elif val == '8':
            path = WORK_ROOT + '/BlendGAN'
            prog = 'blendgan_demo.py --source_image ./pic/66.jpg'

        elif val == '9':
            path = WORK_ROOT + '/stylegan3-editing'
            prog = 'stylegan3.py'

        elif val == '10':
            path = WORK_ROOT + '/stylegan3-editing'
            prog = 'stylegan3_clip.py'

        else:
            loopf = False
            break

        os.chdir(path)
        command = f'python {prog}' if gpu_d or val == '3' or val == '4'  or val == '6' or val == '7'  else f'python {prog} --cpu'
        os.system(command)


# main関数エントリーポイント(実行開始)
if __name__ == '__main__':
    args = sys.argv
    if len(args) > 1 and args[1] == '--cpu':
        gpu_d = False

    device = 'GPU' if gpu_d else 'CPU'

    print(CYAN + '\n------------------------------')
    print(f'|  AI sample program ({device})   |  Ver 0.02')
    print('------------------------------' + NOCOLOR)

    main(gpu_d)

    print(CYAN + '------------------------------')
    print('|    program end..           |')
    print('------------------------------\n' + NOCOLOR)

更新履歴†

2024/08/27 初版
2024/09/21 Ver 0.02 対応
2024/11/04 新規ページへ移行

参考資料†

Powershell
- とほほのPowerShell入門
- PowerShellって何だ？　バッチから乗り換えた話

Other