Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
S315110032
handpose_x
提交
da7ef089
handpose_x
项目概览
S315110032
/
handpose_x
与 Fork 源项目一致
Fork自
Eric.Lee2021 / handpose_x
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
handpose_x
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
提交
da7ef089
编写于
5月 11, 2021
作者:
Eric.Lee2021
🚴🏻
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Update loss.py
上级
6aa7d7bb
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
0 addition
and
33 deletion
+0
-33
loss/loss.py
loss/loss.py
+0
-33
未找到文件。
loss/loss.py
浏览文件 @
da7ef089
...
...
@@ -38,36 +38,3 @@ def wing_loss(landmarks, labels, w=10., epsilon=2.):
def
got_total_wing_loss
(
output
,
crop_landmarks
):
loss
=
wing_loss
(
output
,
crop_landmarks
)
return
loss
'''
AdaptiveWingLoss
'''
class
AdaptiveWingLoss
(
nn
.
Module
):
def
__init__
(
self
,
omega
=
14
,
theta
=
0.5
,
epsilon
=
1
,
alpha
=
2.1
):
super
(
AdaptiveWingLoss
,
self
).
__init__
()
self
.
omega
=
omega
self
.
theta
=
theta
self
.
epsilon
=
epsilon
self
.
alpha
=
alpha
def
forward
(
self
,
pred
,
target
):
'''
:param pred: BxNxHxH
:param target: BxNxHxH
:return:
'''
y
=
target
y_hat
=
pred
delta_y
=
(
y
-
y_hat
).
abs
()
delta_y1
=
delta_y
[
delta_y
<
self
.
theta
]
delta_y2
=
delta_y
[
delta_y
>=
self
.
theta
]
y1
=
y
[
delta_y
<
self
.
theta
]
y2
=
y
[
delta_y
>=
self
.
theta
]
loss1
=
self
.
omega
*
torch
.
log
(
1
+
torch
.
pow
(
delta_y1
/
self
.
omega
,
self
.
alpha
-
y1
))
A
=
self
.
omega
*
(
1
/
(
1
+
torch
.
pow
(
self
.
theta
/
self
.
epsilon
,
self
.
alpha
-
y2
)))
*
(
self
.
alpha
-
y2
)
*
(
torch
.
pow
(
self
.
theta
/
self
.
epsilon
,
self
.
alpha
-
y2
-
1
))
*
(
1
/
self
.
epsilon
)
C
=
self
.
theta
*
A
-
self
.
omega
*
torch
.
log
(
1
+
torch
.
pow
(
self
.
theta
/
self
.
epsilon
,
self
.
alpha
-
y2
))
loss2
=
A
*
delta_y2
-
C
return
(
loss1
.
sum
()
+
loss2
.
sum
())
/
(
len
(
loss1
)
+
len
(
loss2
))
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录