vhad/body-pose-animation
1
0
Fork 0
mirror of https://github.com/gosticks/body-pose-animation.git synced 2025-10-16 11:45:42 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add body mean loss

Browse Source
This commit is contained in:
Wlad 2021-02-05 22:15:57 +01:00
parent a820443206
commit 2a9f7f1676
2 changed files with 22 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
config.yaml
View File

@ -19,15 +19,18 @@ camera:
patience: 10 patience: 10
optimizer: Adam optimizer: Adam
pose: pose:
lr: 0.01 lr: 0.05
optimizer: Adam # currently supported Adam, LBFGS optimizer: Adam # currently supported Adam, LBFGS
iterations: 300 iterations: 300
useCameraIntrinsics: false useCameraIntrinsics: false
bodyMeanLoss:
enabled: false
weight: 0.001
bodyPrior: bodyPrior:
enabled: true enabled: true
weight: 4.0 weight: 0.5
anglePrior: anglePrior:
enabled: false enabled: true
weight: 0.01 weight: 0.01
# idxWeights: # idxWeights:
# - 0.5 # - 0.5

18
train_pose.py
View File

@ -30,9 +30,14 @@ def train_pose(
useConfWeights=True, useConfWeights=True,
patience=10, patience=10,
body_prior_weight=2, body_prior_weight=2,
angle_prior_weight=0.5 angle_prior_weight=0.5,
body_mean_loss=False,
body_mean_weight=0.01
): ):
print("[pose] starting training")
print("[pose] ")
loss_layer = torch.nn.MSELoss() loss_layer = torch.nn.MSELoss()
# setup keypoint data # setup keypoint data
@ -95,9 +100,16 @@ def train_pose(
else: else:
loss = loss_layer(points, keypoints) loss = loss_layer(points, keypoints)
if body_mean_loss:
# apply pose prior loss.
# poZ.pow(2).sum() * body_prior_weight
loss = loss + (pose_layer.body_pose -
pose_extra).pow(2).sum() * body_mean_weight
if useBodyPrior: if useBodyPrior:
# apply pose prior loss. # apply pose prior loss.
loss = loss + poZ.pow(2).sum() * body_prior_weight loss = loss + poZ.pow(2).sum() * body_prior_weight
if useAnglePrior: if useAnglePrior:
loss = loss + \ loss = loss + \
angle_prior_layer(pose_layer.body_pose) * angle_prior_weight angle_prior_layer(pose_layer.body_pose) * angle_prior_weight
@ -187,5 +199,7 @@ def train_pose_with_conf(
optimizer_type=config['pose']['optimizer'], optimizer_type=config['pose']['optimizer'],
iterations=config['pose']['iterations'], iterations=config['pose']['iterations'],
body_prior_weight=config['pose']['bodyPrior']['weight'], body_prior_weight=config['pose']['bodyPrior']['weight'],
angle_prior_weight=config['pose']['anglePrior']['weight'] angle_prior_weight=config['pose']['anglePrior']['weight'],
body_mean_loss=config['pose']['bodyMeanLoss']['enabled'],
body_mean_weight=config['pose']['bodyMeanLoss']['weight']
) )