Instructions to use Eshwar-2123/diffusion_pick_place_clean with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- LeRobot
How to use Eshwar-2123/diffusion_pick_place_clean with LeRobot:
- Notebooks
- Google Colab
- Kaggle
Create README.md
Browse files
README.md
ADDED
|
@@ -0,0 +1,34 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Diffusion Policy for Pick-and-Place (SO-100)
|
| 2 |
+
|
| 3 |
+
## Overview
|
| 4 |
+
This model is a Diffusion Policy trained using Hugging Face LeRobot for a single-task pick-and-place manipulation problem.
|
| 5 |
+
|
| 6 |
+
## Robot
|
| 7 |
+
- SO-100 follower arm
|
| 8 |
+
- Front camera
|
| 9 |
+
- Wrist camera
|
| 10 |
+
|
| 11 |
+
## Task
|
| 12 |
+
Pick the blue cube from one of four predefined locations and place it into a fixed target box.
|
| 13 |
+
|
| 14 |
+
## Dataset
|
| 15 |
+
- 40 teleoperated demonstrations
|
| 16 |
+
- Dual-camera observations
|
| 17 |
+
- Dataset:
|
| 18 |
+
Eshwar-2123/pick_place_blue_cube_box_clean_v1_20260706_125359
|
| 19 |
+
|
| 20 |
+
## Training
|
| 21 |
+
- Policy: Diffusion Policy
|
| 22 |
+
- Steps: 30,000
|
| 23 |
+
- Batch size: 16
|
| 24 |
+
- GPU: NVIDIA RTX 5080
|
| 25 |
+
|
| 26 |
+
## Results
|
| 27 |
+
- Lowest validation loss: 0.037 (10k steps)
|
| 28 |
+
- Final training loss: 0.0036
|
| 29 |
+
- Final validation loss: 0.1303
|
| 30 |
+
|
| 31 |
+
## Usage
|
| 32 |
+
|
| 33 |
+
```python
|
| 34 |
+
from lerobot.policies import DiffusionPolicy
|