[百度飞浆]paddlehub--风格融合https://blog.csdn.net/kobeyu652453/arti

blog.csdn.net/kobeyu65245…

pip

pip install opencv-python
python -m pip install paddlepaddle -i https://mirror.baidu.com/pypi/simple
pip install -i https://mirror.baidu.com/pypi/simple paddlehub

py代码

#!/usr/bin/env python3
# -*- coding: utf-8 -*-


import cv2
import paddlehub as hub

# 加载模型库
stylepro_artistic = hub.Module(name="stylepro_artistic")

# 进行风格迁移
im = stylepro_artistic.style_transfer(
    images=[{
            # 原图
            'content': cv2.imread("baitian.jpg"),
            # 风格图
            'styles': [cv2.imread("yewan.jpg")]
        }],
    # 透明度
    alpha = 0.7
)
# 从返回的数据中获取图片的ndarray对象
im = im[0]['data']
# 保存结果图片
cv2.imshow('result',im)
cv2.imwrite('result.jpg', im)
cv2.waitKey()
cv2.destroyAllWindows()