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

fix indexed access

Browse Source
This commit is contained in:
Wlad 2021-02-08 14:11:23 +01:00
parent dbcab9cffc
commit 809f60bfae
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
example_fit_anim.py
View File

@ -13,8 +13,8 @@ from modules.camera import SimpleCamera
from train_pose import train_pose_with_conf from train_pose import train_pose_with_conf
from utils.general import rename_files, get_new_filename from utils.general import rename_files, get_new_filename
START_IDX = 150 # starting index of the frame to optimize for START_IDX = 0 # starting index of the frame to optimize for
FINISH_IDX = 200 # choose a big number to optimize for all frames in samples directory FINISH_IDX = 50 # choose a big number to optimize for all frames in samples directory
# if False, only run already saved animation without optimization # if False, only run already saved animation without optimization
RUN_OPTIMIZATION = True RUN_OPTIMIZATION = True
@ -60,8 +60,8 @@ joints = model_out.joints.detach().cpu().numpy().squeeze()
Optimization part without visualization Optimization part without visualization
''' '''
if RUN_OPTIMIZATION: if RUN_OPTIMIZATION:
for idx in trange(100, desc='Optimizing'): for idx in trange(FINISH_IDX, desc='Optimizing'):
idx = START_IDX + idx
init_keypoints, init_joints, keypoints, conf, est_scale, r, img_path = setup_training( init_keypoints, init_joints, keypoints, conf, est_scale, r, img_path = setup_training(
model=model, model=model,
renderer=True, renderer=True,