pytorch 液态算法实现瘦脸效果
更新时间：2022年01月18日 17:18:03 作者：watersink
在PS中，我们可以利用液化工具对人像进行形变处理，例如瘦脸、瘦腿、放大眼睛等一系列的常规操作。今天我们来了解一下这些操作的算法原理，并用pytorch来实现瘦脸效果
论文：Interactive Image Warping（1993年Andreas Gustafsson）
算法思路：
假设当前点为（x,y）,手动指定变形区域的中心点为C(cx,cy)，变形区域半径为r，手动调整变形终点（从中心点到某个位置M）为M（mx,my），变形程度为strength，当前点对应变形后的目标位置为U。变形规律如下，
- 圆内所有像素均沿着变形向量的方向发生偏移
- 距离圆心越近，变形程度越大
- 距离圆周越近，变形程度越小，当像素点位于圆周时，该像素不变形
- 圆外像素不发生偏移
其中，x是圆内任意一点坐标，c是圆心点，rmax为圆心半径，m为调整变形的终点，u为圆内任意一点x对应的变形后的位置。
对上面公式进行改进，加入变形程度控制变量strength，改进后瘦脸公式如下，
优缺点：
优点：形变思路简单直接
缺点：
- 局部变形算法，只能基于一个中心点，向另外一个点的方向啦。如果想多个点一起拉伸，只能每个点分别做一次液化，通过针对多个部位多次液化来实现。
- 单点拉伸的变形，可以实现瘦脸的效果，但是效果自然度有待提升。
代码实现：
import cv2
import math
import numpy as np
def localTranslationWarpFastWithStrength(srcImg, startX, startY, endX, endY, radius, strength):
ddradius = float(radius * radius)
copyImg = np.zeros(srcImg.shape, np.uint8)
copyImg = srcImg.copy()
maskImg = np.zeros(srcImg.shape[:2], np.uint8)
cv2.circle(maskImg, (startX, startY), math.ceil(radius), (255, 255, 255), -1)
K0 = 100/strength
# 计算公式中的|m-c|^2
ddmc_x = (endX - startX) * (endX - startX)
ddmc_y = (endY - startY) * (endY - startY)
H, W, C = srcImg.shape
mapX = np.vstack([np.arange(W).astype(np.float32).reshape(1, -1)] * H)
mapY = np.hstack([np.arange(H).astype(np.float32).reshape(-1, 1)] * W)
distance_x = (mapX - startX) * (mapX - startX)
distance_y = (mapY - startY) * (mapY - startY)
distance = distance_x + distance_y
K1 = np.sqrt(distance)
ratio_x = (ddradius - distance_x) / (ddradius - distance_x + K0 * ddmc_x)
ratio_y = (ddradius - distance_y) / (ddradius - distance_y + K0 * ddmc_y)
ratio_x = ratio_x * ratio_x
ratio_y = ratio_y * ratio_y
UX = mapX - ratio_x * (endX - startX) * (1 - K1/radius)
UY = mapY - ratio_y * (endY - startY) * (1 - K1/radius)
np.copyto(UX, mapX, where=maskImg == 0)
np.copyto(UY, mapY, where=maskImg == 0)
UX = UX.astype(np.float32)
UY = UY.astype(np.float32)
copyImg = cv2.remap(srcImg, UX, UY, interpolation=cv2.INTER_LINEAR)
return copyImg
image = cv2.imread("./tests/images/klst.jpeg")
processed_image = image.copy()
startX_left, startY_left, endX_left, endY_left = 101, 266, 192, 233
startX_right, startY_right, endX_right, endY_right = 287, 275, 192, 233
radius = 45
strength = 100
# 瘦左边脸
processed_image = localTranslationWarpFastWithStrength(processed_image, startX_left, startY_left, endX_left, endY_left, radius, strength)
# 瘦右边脸
processed_image = localTranslationWarpFastWithStrength(processed_image, startX_right, startY_right, endX_right, endY_right, radius, strength)
cv2.imwrite("thin.jpg", processed_image)实验效果：
到此这篇关于pytorch 液态算法实现瘦脸效果的文章就介绍到这了,更多相关pytorch 液态算法内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
PyCharm是一款由JetBrains开发的Python集成开发环境（IDE）,它集成了代码编辑器、调试器、版本控制工具和测试工具等功能,下面这篇文章主要给大家介绍了关于pycharm安装包失败的解决方法,需要的朋友可以参考下2023-05-05
python爬取之json、pickle与shelve库的深入讲解
这篇文章主要给大家介绍了关于python爬取之json、pickle与shelve库的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2021-03-03
是否记得，在上一部分，有一讲专门介绍if语句的：从if开始语句的征程。在学习if语句的时候，对python编程的基础知识了解的还不是很多，或许没有做什么太复杂的东西。本讲要对它进行一番复习，通过复习提高一下。如果此前有的东西忘记了，建议首先回头看看前面那讲。2014-10-10
混淆矩阵也称误差矩阵，是表示精度评价的一种标准格式，下面这篇文章主要给大家介绍了关于matplotlib画混淆矩阵与正确率曲线的相关资料，需要的朋友可以参考下2021-06-06
这篇文章主要介绍了Python使用tkinter实现的摇骰子小游戏功能，本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下2020-07-07
这篇文章主要介绍了通过C++学习Python，通过对比分析，让我们能够更好的学习python.2015-01-01
这篇文章主要介绍了Python实用库 PrettyTable 学习笔记,结合实例形式分析了Python表格操作库PrettyTable的安装、使用技巧与相关注意事项,需要的朋友可以参考下2019-08-08
mitmproxy 是一款工具，也可以说是 python 的一个包，在命令行操作的工具。这篇文章主要介绍了详解安装mitmproxy以及遇到的坑和简单用法，感兴趣的小伙伴们可以参考一下2019-01-01
这篇文章主要介绍了Python中用sleep()方法操作时间的教程,是Python入门学习中的基础知识,需要的朋友可以参考下2015-05-05
这篇文章主要介绍了Python根据过滤器拆分列表，利用Python代码实现代通过过滤器拆分列表的功能。文章围绕其相关资料展开详细内容,需要的朋友可以参考一下2021-11-11
最新评论