reorder
This commit is contained in:
mike
@@ -1,141 +0,0 @@
|
||||
#!/usr/bin/env python
|
||||
"""
|
||||
orbit_poc.py — 2.5D actor orbit preview proof-of-concept.
|
||||
|
||||
Usage:
|
||||
python orbit_poc.py --input img1.png img2.png ... --output ./output
|
||||
python orbit_poc.py --input ./filmstrip_images/ --output ./output --mode swing --frames 36
|
||||
|
||||
Output:
|
||||
./output/orbit_frames/frame_NNN.png
|
||||
./output/orbit_preview.mp4
|
||||
./output/debug/selected_frame.png
|
||||
./output/debug/actor_rgba.png
|
||||
./output/debug/mask.png
|
||||
./output/debug/depth.png
|
||||
./output/debug/depth_colorized.png
|
||||
"""
|
||||
|
||||
import argparse
|
||||
import glob
|
||||
import os
|
||||
import sys
|
||||
import time
|
||||
|
||||
# Ensure tour-comfy is on the path when running from project root
|
||||
_here = os.path.dirname(os.path.abspath(__file__))
|
||||
if _here not in sys.path:
|
||||
sys.path.insert(0, _here)
|
||||
|
||||
from orbit_module import run_orbit_pipeline
|
||||
|
||||
|
||||
_IMAGE_EXTS = {".png", ".jpg", ".jpeg", ".webp", ".bmp", ".tiff", ".tif"}
|
||||
|
||||
|
||||
def _collect_inputs(raw_inputs: list) -> list:
|
||||
"""Expand dirs and glob patterns; return sorted list of image paths."""
|
||||
paths = []
|
||||
for item in raw_inputs:
|
||||
if os.path.isdir(item):
|
||||
for fname in sorted(os.listdir(item)):
|
||||
if os.path.splitext(fname)[1].lower() in _IMAGE_EXTS:
|
||||
paths.append(os.path.join(item, fname))
|
||||
elif "*" in item or "?" in item:
|
||||
paths.extend(sorted(glob.glob(item)))
|
||||
elif os.path.isfile(item):
|
||||
paths.append(item)
|
||||
else:
|
||||
print(f"[warn] not found: {item}", file=sys.stderr)
|
||||
return paths
|
||||
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(
|
||||
description="Generate a 2.5D orbit preview (depth-card parallax) from actor images."
|
||||
)
|
||||
parser.add_argument(
|
||||
"--input", "-i", nargs="+", required=True,
|
||||
metavar="PATH",
|
||||
help="Input image paths, glob patterns, or directories",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--output", "-o", default="./output",
|
||||
metavar="DIR",
|
||||
help="Output directory (default: ./output)",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--frames", "-f", type=int, default=36,
|
||||
help="Number of animation frames (default: 36)",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--fps", type=int, default=24,
|
||||
help="Output video framerate (default: 24)",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--parallax", "-p", type=float, default=0.08,
|
||||
help="Parallax strength 0–1, fraction of image width (default: 0.08)",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--angle", "-a", type=float, default=35.0,
|
||||
help="Maximum orbit angle in degrees for swing mode (default: 35)",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--mode", "-m", choices=["swing", "orbit"], default="swing",
|
||||
help="'swing' = left↔right loop (default), 'orbit' = full 360°",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--no-debug", action="store_true",
|
||||
help="Skip writing debug output files",
|
||||
)
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
image_paths = _collect_inputs(args.input)
|
||||
if not image_paths:
|
||||
print("[error] No input images found.", file=sys.stderr)
|
||||
sys.exit(1)
|
||||
|
||||
print(f"[orbit] {len(image_paths)} input image(s)")
|
||||
for p in image_paths[:5]:
|
||||
print(f" {p}")
|
||||
if len(image_paths) > 5:
|
||||
print(f" ... ({len(image_paths) - 5} more)")
|
||||
|
||||
print(f"[orbit] output dir : {os.path.abspath(args.output)}")
|
||||
print(f"[orbit] frames={args.frames} fps={args.fps} mode={args.mode} "
|
||||
f"parallax={args.parallax} angle±{args.angle}°")
|
||||
|
||||
# Use the first image as the primary input (CLI can pass multiple; first = best choice)
|
||||
primary = image_paths[0]
|
||||
if len(image_paths) > 1:
|
||||
print(f"[orbit] using first image as primary: {primary}")
|
||||
print(f" (pass a single image or the specific frame you want)")
|
||||
|
||||
t0 = time.perf_counter()
|
||||
result = run_orbit_pipeline(
|
||||
image_path=primary,
|
||||
output_dir=args.output,
|
||||
n_frames=args.frames,
|
||||
parallax_strength=args.parallax,
|
||||
mode=args.mode,
|
||||
fps=args.fps,
|
||||
max_angle_deg=args.angle,
|
||||
debug=not args.no_debug,
|
||||
)
|
||||
elapsed = time.perf_counter() - t0
|
||||
|
||||
print(f"\n[orbit] done in {elapsed:.1f}s")
|
||||
print(f" actor : {result['actor_path']}")
|
||||
print(f" has alpha : {result['has_alpha']}")
|
||||
print(f" has bg plate : {result['has_bg']}")
|
||||
print(f" frames dir : {result['frames_dir']} ({result['n_frames']} PNGs)")
|
||||
print(f" video : {result['video_path']}")
|
||||
if not args.no_debug:
|
||||
print(f" debug dir : {result['debug_dir']}")
|
||||
if not result['has_alpha']:
|
||||
print(f"\n TIP: Use 'No BG' on this image first for a much better orbit effect.")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user