Add TorchCodec video backend

2026-08-23 16:41:05 +08:00
parent b875a7512e
commit c844673984
5 changed files with 112 additions and 17 deletions
--- a/extract_ego4d_batch.py
+++ b/extract_ego4d_batch.py
@@ -122,7 +122,7 @@ def parse_args(argv: Sequence[str] | None = None) -> argparse.Namespace:
    parser.add_argument("--frame-cache-size", type=int, default=64)
    parser.add_argument(
        "--decoder",
-        choices=("decord", "pynvcodec"),
+        choices=("decord", "torchcodec", "pynvcodec"),
        default="decord",
    )
    parser.add_argument("--decoder-gpu-id", type=int, default=0)