t

Author: Your Name

.gitignore

@@ -140,4 +140,5 @@ temp.gif
 *.mkv
 *.csv
 *.pdf
-.claude/
+.claude/
+*.mp4

examples/droid/droid_downloader.py

@@ -378,8 +378,8 @@ def download_droid_dataset(
     try:
         # Load TFDS dataset
         print("Loading DROID dataset from TFDS...")
-        # ds = tfds.load("droid", data_dir="gs://gresearch/robotics", split="train")
-        ds = tfds.load("droid_100", data_dir="/root/droid-example", split="train")
+        ds = tfds.load("droid", data_dir="gs://gresearch/robotics", split="train")
+        # ds = tfds.load("droid_100", data_dir="/root/droid-example", split="train")
 
         # First pass: Extract episode metadata from TFDS (no Ray)
         print("Extracting episode metadata from TFDS...")
@@ -546,7 +546,7 @@ def download_droid_dataset(
     parser = argparse.ArgumentParser()
     parser.add_argument("--output_dir", default="./droid_downloaded_data", 
                         help="Directory to save downloaded data")
-    parser.add_argument("--num_episodes", type=int, default=100,
+    parser.add_argument("--num_episodes", type=int, default=3000,
                         help="Number of episodes to download")
     parser.add_argument("--num_workers", type=int, default=64,
                         help="Number of parallel workers")

examples/droid/droid_to_robodm.py

@@ -548,15 +548,15 @@ def convert_single_trajectory(traj_dir: str, output_dir: str) -> Tuple[bool, str
 if __name__ == "__main__":
     # Example usage
     processor = DROIDProcessor()
-    output_dir = "./robodm_trajectories"
+    output_dir = "/home/kych/robodm/robodm_trajectories"
 
     try:
         # Parallel download and conversion with 300 success + 100 failure trajectories
         print("Starting parallel download and conversion...")
         successful_paths = processor.download_sample_trajectories(
             output_dir=output_dir, 
-            num_success=50, 
-            num_failure=50
+            num_success=500, 
+            num_failure=500
         )
 
         print(f"\nSuccessfully processed {len(successful_paths)} trajectories:")

examples/droid/droid_vlm_demo.py

@@ -577,7 +577,7 @@ def main():
 
     # Configuration
     parser = argparse.ArgumentParser(description="Run the DROID VLM demo")
-    parser.add_argument("--data_dir", type=str, default="./robodm_trajectories", help="Directory containing RoboDM trajectory files")
+    parser.add_argument("--data_dir", type=str, default="/home/kych/robodm/robodm_trajectories", help="Directory containing RoboDM trajectory files")
     parser.add_argument("--max_trajectories", type=int, default=100, help="Maximum number of trajectories to process")
     args = parser.parse_args()