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

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 ドライブ

drive.google.com

から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 ~

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