mirror of
https://github.com/NousResearch/atropos.git
synced 2026-04-22 16:48:57 +00:00
update readme
This commit is contained in:
parent
332a1025ba
commit
36afc0da59
1 changed files with 19 additions and 1 deletions
|
|
@ -11,4 +11,22 @@ $ pip install pyrender trimesh pyglet matplotlib torch transformers pydantic vll
|
|||
Shared libraries for Ubuntu GL rendering.
|
||||
```sh
|
||||
$ sudo apt-get install libglfw3-dev libgles2-mesa-dev libnvidia-gl-570-server
|
||||
```
|
||||
```
|
||||
|
||||
# Training Data Generation
|
||||
- Use `dataset_scr.py` to create the following directory structure:
|
||||
```
|
||||
dataset/
|
||||
├── stls/
|
||||
│ ├── model_0001.stl
|
||||
│ ├── model_0002.stl
|
||||
│ └── ...
|
||||
├── images/
|
||||
│ ├── model_0001.png
|
||||
│ ├── model_0002.png
|
||||
│ └── ...
|
||||
└── labels.json
|
||||
```
|
||||
- Use `render_stl.py` to generate images from STL files.
|
||||
- Use `llm_label.py` to label the STL and image files.
|
||||
- Use `prepare_push_hf_dataset.py` to push the dataset to Hugging Face.
|
||||
Loading…
Add table
Add a link
Reference in a new issue