split config for lbfgs and adam

This commit is contained in:
Wlad 2021-02-09 10:48:21 +01:00
parent f0205bf56b
commit 66d5360fe1
2 changed files with 70 additions and 7 deletions

View File

@ -0,0 +1,59 @@
output:
rootDir: "./tests/09-02-21"
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: ./output
personId: 0
sampleImageFormat: "input_%%%%%%%%%%%%_rendered.png"
sampleNameFormat: "input_%%%%%%%%%%%%_keypoints.json"
sampleCoords: !!python/tuple [1080, 1080]
camera:
lr: 0.001
patience: 10
optimizer: Adam
pose:
# device: cuda
lr: 0.05
optimizer: LBFGS # currently supported Adam, LBFGS
iterations: 15
useCameraIntrinsics: true
bodyMeanLoss:
enabled: false
weight: 0.1
bodyPrior:
enabled: true
weight: 4.0
anglePrior:
enabled: true
weight: 0.0001
angleSumLoss:
enabled: true
weight: 0.2
confWeights:
enabled: false
vposerPath: "./vposer_v1_0"
temporal:
enabled: true
iterations: 4
lr: 0.02
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

View File

@ -1,6 +1,6 @@
output: output:
rootDir: "./results/" rootDir: "./tests/09-02-21"
prefix: result- prefix: ""
smpl: smpl:
modelRootDir: ./models modelRootDir: ./models
type: smplx # possible options smplx and smpl type: smplx # possible options smplx and smpl
@ -19,10 +19,10 @@ camera:
patience: 10 patience: 10
optimizer: Adam optimizer: Adam
pose: pose:
device: cuda # device: cuda
lr: 0.5 lr: 0.01
optimizer: LBFGS # currently supported Adam, LBFGS optimizer: Adam # currently supported Adam, LBFGS
iterations: 10 iterations: 150
useCameraIntrinsics: true useCameraIntrinsics: true
bodyMeanLoss: bodyMeanLoss:
enabled: false enabled: false
@ -35,10 +35,14 @@ pose:
weight: 0.01 weight: 0.01
angleSumLoss: angleSumLoss:
enabled: false enabled: false
weight: 0.01 weight: 0.02
confWeights: confWeights:
enabled: false enabled: false
vposerPath: "./vposer_v1_0" vposerPath: "./vposer_v1_0"
temporal:
enabled: true
iterations: 30
lr: 0.01
preview: preview:
enable: true, enable: true,
keypoins: keypoins: