Merge branch #roll-ai/vfx-3' into 'roll-ai/vfx-4'

gradio_epic_only.py

@@ -142,15 +142,19 @@ def anchor_generation(
     init_theta,
     init_phi,
     reverse_effect,
-    extended_effect
+    extended_effect,
+    centered_motion
 ):
     cross_frames = 3
     if extended_effect:
         camera = "extended"
         num_frames = 98 - cross_frames
+    elif centered_motion:
+        camera = "centered"
     else:
         camera = "target"
 
+        
     temp_input_path = f"/app/{Unique_identifier}/input.mp4"
     global RENDER_PATH, CROPS_PATH
 
@@ -250,7 +254,7 @@ def anchor_generation(
         "--init_dy", str(init_dy),
         "--init_dz", str(init_dz),
         "--init_theta", str(init_theta),
-        "--init_phi", str(init_phi),
+        "--init_phi", str(init_phi)
     ]
 
     print("🚀 Starting subprocess inference...")

gradio_server.py

@@ -87,7 +87,7 @@ def run_anchor_gen(
     depth_guidance_scale, window_size, overlap, max_res, load_size,
     sample_size, depth_size, seed_input, aspect_ratio_inputs,
     init_dx, init_dy, init_dz, init_theta, init_phi,
-    reverse_effect, extended_effect
+    reverse_effect, extended_effect, centered_motion
 ):
     if not unique_identifier:
         unique_identifier = str(uuid.uuid4())
@@ -136,7 +136,8 @@ def run_anchor_gen(
         init_theta=init_theta,
         init_phi=init_phi,
         reverse_effect=reverse_effect,
-        extended_effect=extended_effect
+        extended_effect=extended_effect,
+        centered_motion=centered_motion
     )
     
     # Save logs, captions, frame_shape
@@ -413,6 +414,7 @@ with gr.Blocks(title="Camera vfx Server") as demo:
                     anchor_phi = gr.Number(label="Init Phi", value=0.0)
                 anchor_reverse = gr.Checkbox(label="Reverse Effect", value=False)
                 anchor_extended = gr.Checkbox(label="Extended Effect", value=False)
+                centered_motion = gr.Checkbox(label="Centered Motion", value=False)
         
         anchor_btn = gr.Button("Generate Anchor")
         anchor_out = gr.File(label="Anchor ZIP Output")
@@ -427,7 +429,7 @@ with gr.Blocks(title="Camera vfx Server") as demo:
                 anchor_depth_guidance, anchor_window, anchor_overlap, anchor_max_res, anchor_load_size,
                 anchor_sample_size, anchor_depth_size, anchor_seed, anchor_aspect,
                 anchor_dx, anchor_dy, anchor_dz, anchor_theta, anchor_phi,
-                anchor_reverse, anchor_extended
+                anchor_reverse, anchor_extended, centered_motion
             ],
             outputs=[anchor_out],
             api_name="anchor"

inference/v2v_data/demo.py

@@ -684,7 +684,7 @@ class GetAnchorVideos:
             .unsqueeze(0)
         )
         
-        if opts.camera == "extended":
+        if opts.camera == "extended" or opts.camera == "centered":
             print("Using reverse camera trajectory")
             
             dtheta, dphi, dr, dx, dy = opts.target_pose

inference/v2v_data/models/utils.py

@@ -402,6 +402,7 @@ def generate_traj_specified_fast(c2ws_anchor, theta, phi, d_r, d_x, d_y, frame,
 
     c2ws_list = []
 
+
     for th, ph, r, x, y in zip(thetas, phis, rs, xs, ys):
         c2w_new = sphere2pose(
             c2ws_anchor,