Upload 5 files

download_all.sh

@@ -0,0 +1,7 @@
+#!/bin/bash
+if [ "$1" = "" ]
+then
+  echo "Usage: $0 <output-dir>"
+  exit
+fi
+python downloader.py -o $1 -s 010_0050 020_0020 040_0040 050_0130 050_0160 060_0100 060_0130 070_0123 -m rgb depth mask segmentation normal flow_fw flow_bw -t Dense Sparse Mono

download_rgb_dense.sh

@@ -0,0 +1,7 @@
+#!/bin/bash
+if [ "$1" = "" ]
+then
+  echo "Usage: $0 <output-dir>"
+  exit
+fi
+python downloader.py -o $1 -s 010_0050 020_0020 040_0040 050_0130 050_0160 060_0100 060_0130 070_0123 -m rgb -t Dense

download_rgb_mono.sh

@@ -0,0 +1,7 @@
+#!/bin/bash
+if [ "$1" = "" ]
+then
+  echo "Usage: $0 <output-dir>"
+  exit
+fi
+python downloader.py -o $1 -s 010_0050 020_0020 040_0040 050_0130 050_0160 060_0100 060_0130 070_0123 -m rgb -t Mono

download_rgb_sparse.sh

@@ -0,0 +1,7 @@
+#!/bin/bash
+if [ "$1" = "" ]
+then
+  echo "Usage: $0 <output-dir>"
+  exit
+fi
+python downloader.py -o $1 -s 010_0050 020_0020 040_0040 050_0130 050_0160 060_0100 060_0130 070_0123 -m rgb -t Sparse

downloader.py

@@ -0,0 +1,69 @@
+from huggingface_hub import snapshot_download
+import argparse
+
+parser = argparse.ArgumentParser()
+parser.add_argument(
+    "--output-dir", "-o", required=True, help="Output directory for the dataset."
+)
+parser.add_argument(
+    "--scenes", "-s", required=True, nargs="+", help="List of scenes to download.",
+    choices=["010_0050", "020_0020", "040_0040", "050_0130", "050_0160", "060_0100", "060_0130", "070_0123"]
+)
+parser.add_argument(
+    "--modalities", "-m", required=True, nargs="*", help="List of scenes to download.",
+    choices=["rgb", "depth", "mask", "segmentation", "normal", "flow_fw", "flow_bw"]
+)
+parser.add_argument(
+    "--tasks", "-t", required=True, nargs="+", help="Tasks to download.",
+    choices=["Dense", "Sparse", "Mono"]
+)
+
+def main(args):
+    # Collect scenes from multiple repos
+    repo_type = "dataset"
+
+    # Set modalities patterns
+    all_modalities = ["rgb", "depth", "mask", "segmentation", "normal", "flow_fw", "flow_bw"]
+    modalities_exclude = set(all_modalities) - set(args.modalities)
+    exclude_patterns = []
+    for mod in modalities_exclude:
+        if mod == "rgb":
+            exclude_patterns.append("*frame_????.png")
+        elif mod == "depth":
+            exclude_patterns.append("*_depth.npy")
+        elif mod == "mask":
+            exclude_patterns.append("*_dyn_mask.png")
+        elif mod == "flow_fw":
+            exclude_patterns.append("*_flow_fw.npy")
+        elif mod == "flow_bw":
+            exclude_patterns.append("*_flow_bw.npy")
+        elif mod == "segmentation":
+            exclude_patterns.append("*_segmentation.png")
+        elif mod == "normal":
+            exclude_patterns.append("*_normal.png")
+
+    # Set task patterns
+    allow_patterns = []
+    for task in args.tasks:
+        allow_patterns.append(f"*/{task}/*")
+    if "segmentation" not in modalities_exclude:
+        allow_patterns.append("*segmentation.json")
+
+    # Download from respective repos
+    for scene in args.scenes:
+        if scene == "050_0130":
+            repo_id = "charge-benchmark/Charge-050_0130"
+            snapshot_download(
+                repo_id=repo_id,
+                repo_type=repo_type,
+                allow_patterns=allow_patterns,
+                ignore_patterns=exclude_patterns,
+                local_dir=args.output_dir
+            )
+        else:
+            print(f"Scene {scene} not available yet.")
+        
+
+if __name__ == "__main__":
+    args = parser.parse_args()
+    main(args)