git clone https://github.com/mit-han-lab/foreact
cd foreact
bash environment_setup.sh foreact
Download the pretrained weights and prepare your own real-world data (or use our processed real-world data). Update the relevant paths in the config you want to run (configs/finetune.yaml for single-view, configs/finetune_multiview.yaml for multi-view), then launch:
### Single-view
bash scripts/run_finetune.sh finetune.yaml
### Multi-view
bash scripts/run_finetune.sh finetune_multiview.yaml
### CLI (single-view)
python app_cli.py --checkpoint_path path/to/model --prompt "" \
--input_image primary=path/to/image --output_dir ./results
### CLI (multi-view)
python app_cli.py --checkpoint_path path/to/model --prompt "" \
--input_image primary=path/to/img1 wrist_left=path/to/img2 wrist_right=path/to/img3 \
--output_dir ./results
### Gradio (auto-adapts to the checkpoint's view configuration)
python app.py --checkpoint_path path/to/model
We provide examples regarding policy training in ./third-party/lerobot.
Thanks to metaquery, diffusers, lerobot for the wonderful open-source codebase.
4 commits
2 commits
Python
98.2%
Shell
1.8%
git clone https://github.com/mit-han-lab/foreact
cd foreact
bash environment_setup.sh foreact
Download the pretrained weights and prepare your own real-world data (or use our processed real-world data). Update the relevant paths in the config you want to run (configs/finetune.yaml for single-view, configs/finetune_multiview.yaml for multi-view), then launch:
### Single-view
bash scripts/run_finetune.sh finetune.yaml
### Multi-view
bash scripts/run_finetune.sh finetune_multiview.yaml
### CLI (single-view)
python app_cli.py --checkpoint_path path/to/model --prompt "" \
--input_image primary=path/to/image --output_dir ./results
### CLI (multi-view)
python app_cli.py --checkpoint_path path/to/model --prompt "" \
--input_image primary=path/to/img1 wrist_left=path/to/img2 wrist_right=path/to/img3 \
--output_dir ./results
### Gradio (auto-adapts to the checkpoint's view configuration)
python app.py --checkpoint_path path/to/model
We provide examples regarding policy training in ./third-party/lerobot.
Thanks to metaquery, diffusers, lerobot for the wonderful open-source codebase.
4 commits
2 commits
Python
98.2%
Shell
1.8%