diff --git a/example_fit_anim.py b/example_fit_anim.py index 9c5f62c..19d14c8 100644 --- a/example_fit_anim.py +++ b/example_fit_anim.py @@ -50,7 +50,6 @@ joints = model_out.joints.detach().cpu().numpy().squeeze() ''' Optimization part without visualization ''' - if RUN_OPTIMIZATION: while get_next_frame(idx) is not None and idx <= FINISH_IDX: keypoints, confidence, img_path = get_next_frame(idx) @@ -115,7 +114,7 @@ if RUN_OPTIMIZATION: pickle.dump(final_poses, fp) print("Results have been saved to", filename) -# TODO: put into utils, rename file, +# TODO: put into utils, rename file, use current body pose and camera transform for next optimization def replay_animation(file, start_frame=0, end_frame=None, with_background=False, fps=30): r = Renderer() r.start() @@ -128,15 +127,12 @@ def replay_animation(file, start_frame=0, end_frame=None, with_background=False, if end_frame is None: end_frame = len(final_poses) - print(len(final_poses)) - for i in range(start_frame, end_frame): body_pose = final_poses[i][0] camera_transform = final_poses[i][1] - # Changing image is too jerky, because the image has to be removed and added each time - if with_background: + # Changing image is too jerky, because the image has to be removed and added each time pass # img_path = samples_dir + "/" + str(i) + ".png" # if r.get_node("image") is not None: diff --git a/results/result-3.pkl b/results/result-0.pkl similarity index 52% rename from results/result-3.pkl rename to results/result-0.pkl index 7e1cdfc..5ad7e96 100644 Binary files a/results/result-3.pkl and b/results/result-0.pkl differ