Buckets:
| import argparse | |
| import os | |
| from pathlib import Path | |
| def main(): | |
| os.environ.setdefault("MUJOCO_GL", "egl") | |
| from reconstruct_mapanything import main as reconstruct | |
| from replay_mapanything import main as replay | |
| from real2sim import Episode | |
| parser = argparse.ArgumentParser() | |
| parser.add_argument("--task", default="basic_pick_place") | |
| parser.add_argument("--index", type=int, default=0) | |
| parser.add_argument("--output", type=Path, required=True) | |
| parser.add_argument("--device", default="cuda") | |
| parser.add_argument("--stride", type=int, default=5) | |
| parser.add_argument("--max-frames", type=int, default=128) | |
| parser.add_argument("--tracks", type=Path) | |
| args = parser.parse_args() | |
| episode = Episode.load(args.task, args.index) | |
| reconstruct( | |
| [ | |
| str(episode.video_path), | |
| "--output", | |
| str(args.output), | |
| "--device", | |
| args.device, | |
| "--stride", | |
| str(args.stride), | |
| "--max-frames", | |
| str(args.max_frames), | |
| "--preview", | |
| ] | |
| ) | |
| replay_args = [str(args.output), "--task", args.task, "--index", str(args.index)] | |
| if args.tracks: | |
| replay_args.extend(["--tracks", str(args.tracks)]) | |
| replay(replay_args) | |
| if __name__ == "__main__": | |
| main() | |
Xet Storage Details
- Size:
- 1.33 kB
- Xet hash:
- ecbb91c9bf318cb177d0734c29f90e61f0c08114b8bb30b4e0a5673b8d8a7612
·
Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.