Update merge.py

Modifications
Introduced the argparse module to parse command-line arguments
Converted previously hardcoded source folder paths, output folder path, max_dim, and FPS values to configurable command-line parameters
Important Note
Note: When merging datasets, the FPS for each collected task must be identical to ensure data synchronization and consistency

lerobot/scripts/merge.py

@@ -1292,13 +1292,21 @@ def merge_datasets(
 
 
 if __name__ == "__main__":
-    # Define source folders and output folder
+    # Set up argument parser
-    source_folders = ["/path/to/put_plastic_to_box/", "/path/to/put_metal_to_box/", "/path/to/Find_blue_box"]
+    parser = argparse.ArgumentParser(description='Merge datasets from multiple sources.')
     
-    output_folder = "/data1/realman/lerobot_merged_test/"
+    # Add arguments
+    parser.add_argument('--sources', nargs='+', required=True,
+                        help='List of source folder paths')
+    parser.add_argument('--output', required=True,
+                        help='Output folder path')
+    parser.add_argument('--max_dim', type=int, default=32,
+                        help='Maximum dimension (default: 32)')
+    parser.add_argument('--fps', type=int, default=20,
+                        help='Your datasets FPS (default: 20)')
     
-    # 设置默认FPS
+    # Parse arguments
-    default_fps = 20
+    args = parser.parse_args()
     
-    # Merge the datasets
+    # Use parsed arguments
-    merge_datasets(source_folders, output_folder, max_dim=32, default_fps=default_fps)
+    merge_datasets(args.sources, args.output, max_dim=args.max_dim, default_fps=args.fps)