Update scripts/run_preprocessing.py

scripts/run_preprocessing.py

@@ -1,54 +1,42 @@
-import subprocess
-import sys
-import os
+import subprocess, sys, os
 import tyro
-
 from pixel3dmm import env_paths
 
 def run_and_check(cmd, cwd=None):
-    """
-    Run a command (list of args) in an optional cwd.
-    Raises CalledProcessError (with stdout/stderr) on failure.
-    """
     print(f"> {' '.join(cmd)}  (in {cwd or os.getcwd()})")
+    # stream logs live
     result = subprocess.run(
         cmd,
         cwd=cwd,
-        stdout=subprocess.PIPE,
-        stderr=subprocess.PIPE,
-        text=True,
-        check=True,            # will raise on non-zero exit
+        check=True,
     )
-    print(result.stdout)      # print normal output
     return result
 
 def main(video_or_images_path: str):
-    # derive video name
-    if os.path.isdir(video_or_images_path):
-        vid_name = os.path.basename(video_or_images_path)
-    else:
-        vid_name, _ = os.path.splitext(os.path.basename(video_or_images_path))
+    vid_name = (
+        os.path.basename(video_or_images_path)
+        if os.path.isdir(video_or_images_path)
+        else os.path.splitext(os.path.basename(video_or_images_path))[0]
+    )
+
+    SCRIPTS = os.path.join(env_paths.CODE_BASE, "scripts")
+    MICA    = os.path.join(env_paths.CODE_BASE, "src", "pixel3dmm", "preprocessing", "MICA")
 
     try:
-        # cropping
         run_and_check(
-            ["python", "run_cropping.py", "--video_or_images_path", video_or_images_path],
-            cwd=os.path.join(env_paths.CODE_BASE, "scripts")
+            [sys.executable, "-u", "run_cropping.py", "--video_or_images_path", video_or_images_path],
+            cwd=SCRIPTS,
         )
-        # MICA preprocessing
         run_and_check(
-            ["python", "demo.py", "-video_name", vid_name],
-            cwd=os.path.join(env_paths.CODE_BASE, "src", "pixel3dmm", "preprocessing", "MICA")
+            [sys.executable, "-u", "demo.py", "-video_name", vid_name],
+            cwd=MICA,
         )
-        # face segmentation
         run_and_check(
-            ["python", "run_facer_segmentation.py", "--video_name", vid_name],
-            cwd=os.path.join(env_paths.CODE_BASE, "scripts")
+            [sys.executable, "-u", "run_facer_segmentation.py", "--video_name", vid_name],
+            cwd=SCRIPTS,
         )
-
     except subprocess.CalledProcessError as e:
-        # Print the error, including stdout/stderr
-        print(f"ERROR: Command {e.cmd!r} exited with {e.returncode}", file=sys.stderr)
+        print(f"ERROR: {e.cmd!r} exited with {e.returncode}", file=sys.stderr)
         print("---- STDOUT ----", file=sys.stderr)
         print(e.stdout, file=sys.stderr)
         print("---- STDERR ----", file=sys.stderr)