[NCDA]一键教你玩转飞桨人像模型

# 从gitee上克隆PaddleGAN代码!git clone https://gitee.com/paddlepaddle/PaddleGAN

# 安装所需安装包 17秒左右%cd /home/aistudio/PaddleGAN/
!pip install -r requirements.txt
%cd /home/aistudio/PaddleGAN/applications/

#安装所需包 首次此安装包需要5分钟31秒484毫秒!pip install dlib 
%cd /home/aistudio/PaddleGAN  
!python setup.py develop
%cd /home/aistudio/PaddleGAN/applications

# 准备第一张图片 只需要修改第三行代码为你的路径即可!python -u tools/styleganv2fitting.py \
       --input_image '/home/aistudio/work/11.png'\
       --need_align \
       --start_lr 0.1 \
       --final_lr 0.025 \
       --latent_level 0 1 2 3 4 5 6 7 8 9 10 11 \
       --step 100 \
       --mse_weight 1 \
       --output_path '/home/aistudio/output/11' \
       --model_type ffhq-config-f \
       --size 512 \
       --style_dim 512 \
       --n_mlp 8 \
       --channel_multiplier 2

# 准备第二张图片 只需要修改第三行代码为你的路径即可!python -u tools/styleganv2fitting.py \
       --input_image '/home/aistudio/work/22.png'\
       --need_align \
       --start_lr 0.1 \
       --final_lr 0.025 \
       --latent_level 0 1 2 3 4 5 6 7 8 9 10 11 \
       --step 100 \
       --mse_weight 1 \
       --output_path '/home/aistudio/output/22' \
       --model_type ffhq-config-f \
       --size 512 \
       --style_dim 512 \
       --n_mlp 8 \
       --channel_multiplier 2

!python -u tools/styleganv2mixing.py \
       --latent1 '/home/aistudio/output/11/dst.fitting.npy' \
       --latent2 '/home/aistudio/output/22/dst.fitting.npy' \
       --weights \                 0.5 0.5 0.5 0.5 0.5 0.5 \                 0.5 0.5 0.5 0.5 0.5 0.5 \                 0.5 0.5 0.5 0.5 0.5 0.5 \
       --output_path '/home/aistudio/mixoutput/11&22' \
       --model_type ffhq-config-f \
       --size 512 \
       --style_dim 512 \
       --n_mlp 8 \
       --channel_multiplier 2

#安装PaddleHub!pip install --upgrade paddlehub -i https://mirror.baidu.com/pypi/simple# 该模型需要gpu跑，所以需要设置下gpu环境%env CUDA_VISIBLE_DEVICES=0%matplotlib inline# 转换为动漫图片import cv2import paddlehub as hub# 多种风格可以选# 宫崎骏风格model1 = hub.Module(name='animegan_v1_hayao_60', use_gpu=True)# 新海诚风格model2 = hub.Module(name='animegan_v2_shinkai_33', use_gpu=True)# 今敏风格model3 = hub.Module(name='animegan_v2_paprika_74', use_gpu=True)# 模型预测result1 = model1.style_transfer(images=[cv2.imread('/home/aistudio/work/11.png')],visualization=True,output_dir='/home/aistudio/output/ani_image_gqj')
result2 = model2.style_transfer(images=[cv2.imread('/home/aistudio/work/11.png')],visualization=True,output_dir='/home/aistudio/output/ani_image_xhc')
result3 = model3.style_transfer(images=[cv2.imread('/home/aistudio/work/11.png')],visualization=True,output_dir='/home/aistudio/output/ani_image_jm')

# 转换为动漫图片 大概10秒import paddleimport cv2import paddlehub as hub

paddle.disable_static()

model = hub.Module(name='U2Net_Portrait')

result = model.Portrait_GEN(paths=['/home/aistudio/work/11.png'],visualization=True,output_dir='/home/aistudio/output/sumiao')

%cd /home/aistudio/PaddleGAN/applications/

# 第一次会比较慢 43秒左右!python -u tools/pixel2style2pixel.py \
       --input_image '/home/aistudio/work/11.png' \
       --output_path   '/home/aistudio/output/bianxing' \
       --model_type ffhq-inversion \
       --seed 233 \
       --size 1024 \
       --style_dim 512 \
       --n_mlp 8 \
       --channel_multiplier 2

# 大概8秒!python -u tools/styleganv2editing.py \
       --latent '/home/aistudio/output/bianxing/dst.npy' \
       --output_path '/home/aistudio/output/bianxing' \
       --model_type ffhq-config-f \
       --size 1024 \
       --style_dim 512 \
       --n_mlp 8 \
       --channel_multiplier 2 \
       --direction_name gender \
       --direction_offset 5

!pip install boost -i https://pypi.tuna.tsinghua.edu.cn/simple/
!pip install CMake -i https://pypi.tuna.tsinghua.edu.cn/simple/ 
!pip install dlib -i https://pypi.tuna.tsinghua.edu.cn/simple/
%cd /home/aistudio/PaddleGAN/

# 大概30秒!python tools/psgan_infer.py \
    --config-file configs/makeup.yaml \
    --source_path  ~/work/22.png   \
    --reference_dir ~/work/ref \
    --evaluate-only