mmd-auto-trace-4

環境構築

CUDA

(base) miu@garnet:~$ nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Wed_Jun__8_16:49:14_PDT_2022
Cuda compilation tools, release 11.7, V11.7.99
Build cuda_11.7.r11.7/compiler.31442593_0

env

conda create --name mat4 python=3.10
conda activate mat4
conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia

export PATH=/home/miu/anaconda3/envs/mat4/bin:$PATH
pip install -r requirements.txt

画像拡大ライブラリ

mkdir resize
wget -P resize https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesrgan-ncnn-vulkan-20220424-ubuntu.zip 
unzip resize/realesrgan-ncnn-vulkan-20220424-ubuntu.zip -d resize

データ配置

mmd-auto-trace-4/data/basicModel_neutral_lbs_10_207_0_v1.0.0.pkl

バイナリの起動

python py/exec_mediapipe.py --video /mnt/e/MMD_E/201805_auto/02/buster/buster.mp4 --output_dir /mnt/e/MMD_E/201805_auto/02/buster/buster_20240425_015307
python py/smooth.py /mnt/e/MMD_E/201805_auto/02/buster/buster_20240425_015307

./dist/mat4 -modelPath=/mnt/c/MMD/mmd-auto-trace-4/configs/pmx/v4_trace_model.pmx -dirPath=/mnt/e/MMD_E/201805_auto/02/buster/buster_20240425_015307

conda remove -n mat4 --all