body-pose-animation/config.yaml
2021-02-07 17:40:13 +01:00

56 lines
1.1 KiB
YAML

output:
rootDir: "./results/"
prefix: result-
smpl:
modelRootDir: ./models
type: smplx # possible options smplx and smpl
gender: MALE # possible options MALE, FEMALE, NEUTRAL
ext: npz
useVposerInit: false
data:
renameFiles: false
rootDir: ./samples
personId: 0
sampleImageFormat: "%%%.png"
sampleNameFormat: "%%%.json"
sampleCoords: !!python/tuple [1080, 1080]
camera:
lr: 0.001
patience: 10
optimizer: Adam
pose:
# device: cuda
lr: 0.01
optimizer: Adam # currently supported Adam, LBFGS
iterations: 100
useCameraIntrinsics: true
bodyMeanLoss:
enabled: false
weight: 0.1
bodyPrior:
enabled: false
weight: 2.0
anglePrior:
enabled: true
weight: 0.1
angleSumLoss:
enabled: true
weight: 0.01
confWeights:
enabled: false
vposerPath: "./vposer_v1_0"
preview:
enable: true,
keypoins:
enable: true,
radius: 0.01
color: 1.0, 0.0, 1.0, 1.0
keypoint_torso:
enable: true,
radius: 0.01
color: 1.0, 0.0, 1.0, 1.0
joints:
enable: true
radius: 0.01
color: 0.0, 0.7, 0.0, 1.0