globalise

app.py

@@ -137,10 +137,12 @@ def make_sample(
     }
     return sample
 
+g_use_factor_guidance = True
+
 pipe = load_ctrlv_pipelines(
     "AnthonyGosselin/Ctrl-Crash",
     use_null_model=True,
-    use_factor_guidance=False
+    use_factor_guidance=g_use_factor_guidance
 )
 
 def run_yolo_sam_on_image(
@@ -214,7 +216,7 @@ def run_one(
         json_path=os.path.join(out_dir, "gt_frames_minimal.json"),
         bbox_mask_frames=bbox_mask_frames,
         action_type=sample["action_type"].unsqueeze(0),
-        use_factor_guidance=True,
+        use_factor_guidance=g_use_factor_guidance,
         guidance=list(guidance),
         video_path2=None,
     )

src/eval/generate_samples.py

@@ -153,8 +153,6 @@ def generate_video_ctrlv(sample, pipeline, video_path="video_out/genvid", json_p
     bbox_images = sample['bbox_images'].unsqueeze(0)
     action_type = sample['action_type'].unsqueeze(0) if action_type is None else action_type
 
-    print(f'final action type {action_type}')
-
     sample['bbox_images'].to(device)
 
     # Save GT frame paths to json file