krish arul meiyappan commited on
Upload CUSTOMER_INTEGRATION_STEPS.md with huggingface_hub
Browse files
CUSTOMER_INTEGRATION_STEPS.md
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Humanoid Brain Integration Steps
|
| 2 |
+
|
| 3 |
+
1) Install: pip install torch torchvision pillow numpy && git clone https://github.com/krishthesmart/humanoid-brain.git && cd humanoid-brain
|
| 4 |
+
2) Download model: https://huggingface.co/theakboysca/humanoid-brain-v1 (save as model.pt)
|
| 5 |
+
3) Test: python -m humanoid_brain.examples.demo_inference --weights model.pt --image /path/to/test_image.jpg --device cpu
|
| 6 |
+
4) Integrate: from humanoid_brain.sdk.inference_api import load_brain; brain=load_brain("model.pt",device="cpu"); decision=brain.decide(image=rgb_frame,robot_state=robot_state,env_state=env_state)
|
| 7 |
+
5) Output: task in [cleaning,cooking,dishwashing,laundry,organizing], sub_goals for motion stack
|