Ubuntu22.04DesktopでAnime2Sketchのインストール記録

AI・機械学習 コンピュータ
AI・機械学習

Anime2Sketchをインストールしたのでその作業ログを残しておきます。

pythonの仮想環境を作成

cd ~/git 
python3 -m venv atosvenv

環境名をatosvenvにしました。

仮想環境の開始

source ~/git/atosvenv/bin/activate

仮想環境の終了要確認

deactivate

必要なパッケージをインストール

python -m pip install --upgrade pip
python -m pip install -U --ignore-installed pip
python -m pip install -U torch torchvision torchaudio numpy --index-url https://download.pytorch.org/whl/cu115
python -m pip install -U tqdm
python -m pip install -U kornia
cd ~/git
git clone --recursive https://github.com/Mukosame/Anime2Sketch
cd Anime2Sketch
モデルのダウンロード
Anime2Sketch - Google ドライブ

からnetG.pthをダウンロードしweightsに保存

仮想環境の開始と入出力ディレクトリを指定した実行用のシェルスクリプト

#!/bin/bash
cd ~
venvFile=`pwd`'/git/atosvenv/bin/activate'
dataRoot='/mnt/workspace'
inputDir=$dataRoot'/inputs'
outputDir=$dataRoot'/outputs'
logfile=$dataRoot'/logs/Anime2Sketch.log'

cd $outputDir
if [ `ls -U1 | wc -l` -ne 0 ]; then
    rm *
fi;
cd $inputDir
if [ `ls -U1 | wc -l` -eq 0 ]; then
    echo 'Not File';
    exit 1
fi;
cd ~/git/Anime2Sketch
source $venvFile
echo '*** Anime2Sketch Start ***' >> $logfile
date >> $logfile
python test.py --dataroot $inputDir --load_size 1024 --output_dir $outputDir >> $logfile
date >> $logfile
echo '*** Anime2Sketch End ***' >> $logfile
deactivate
cd ~

実行環境に合わせて修正する必要あり。

コメント