Update 2026-04-20 00:02:36
This commit is contained in:
parent
a94d20ab15
commit
2c9cde84ca
50
README.md
50
README.md
@ -49,3 +49,53 @@ python -m saqr.robot.bridge --iface eth0 --source realsense --headless
|
|||||||
- [docs/DEPLOY.md](docs/DEPLOY.md) — full deploy + robot setup.
|
- [docs/DEPLOY.md](docs/DEPLOY.md) — full deploy + robot setup.
|
||||||
- [docs/start.md](docs/start.md) — systemd auto-start workflow.
|
- [docs/start.md](docs/start.md) — systemd auto-start workflow.
|
||||||
- [docs/use_case_catalogue.pdf](docs/use_case_catalogue.pdf) — PPE use-case spec.
|
- [docs/use_case_catalogue.pdf](docs/use_case_catalogue.pdf) — PPE use-case spec.
|
||||||
|
|
||||||
|
## Data & Models
|
||||||
|
|
||||||
|
The `data/` and `runtime/` directories are excluded from git (too large).
|
||||||
|
Download them separately before training or running inference.
|
||||||
|
|
||||||
|
### `data/` — dataset and pre-trained weights
|
||||||
|
|
||||||
|
Expected contents:
|
||||||
|
|
||||||
|
```
|
||||||
|
data/
|
||||||
|
dataset/
|
||||||
|
train/{images,labels}/
|
||||||
|
valid/{images,labels}/
|
||||||
|
test/{images,labels}/
|
||||||
|
data.yaml
|
||||||
|
models/
|
||||||
|
saqr_best.pt # Saqr YOLO11n fine-tuned on PPE
|
||||||
|
saqr_last.pt
|
||||||
|
yolo11n.pt # base YOLO11n
|
||||||
|
yolo26n.pt # base YOLO26n
|
||||||
|
```
|
||||||
|
|
||||||
|
Download:
|
||||||
|
|
||||||
|
- **Dataset** (PPE, Roboflow): [testcasque/ppe-detection-qlq3d](https://universe.roboflow.com/testcasque/ppe-detection-qlq3d)
|
||||||
|
Open the Roboflow link → *Download Dataset* → format **YOLOv11** → unzip into `data/dataset/`.
|
||||||
|
- **Base YOLO weights**: [Ultralytics assets releases](https://github.com/ultralytics/assets/releases)
|
||||||
|
Grab `yolo11n.pt` (and optionally `yolo26n.pt`) into `data/models/`.
|
||||||
|
- **Saqr fine-tuned weights** (`saqr_best.pt`, `saqr_last.pt`):
|
||||||
|
Produced by training — see "Training" below. Or request from the maintainer.
|
||||||
|
|
||||||
|
Place everything under `data/` so the tree matches above.
|
||||||
|
|
||||||
|
### `runtime/` — training output (optional)
|
||||||
|
|
||||||
|
Auto-generated when you run training. Not required for inference.
|
||||||
|
Contains confusion matrices, PR curves, batch previews, and the raw weights
|
||||||
|
under `runtime/runs/train/saqr_det/weights/`.
|
||||||
|
|
||||||
|
### Training
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# after placing the dataset in data/dataset/ and base weights in data/models/
|
||||||
|
python -m saqr.apps.train_cli --data data/dataset/data.yaml --weights data/models/yolo11n.pt
|
||||||
|
```
|
||||||
|
|
||||||
|
Outputs land in `runtime/runs/train/saqr_det/`. Copy the best checkpoint to
|
||||||
|
`data/models/saqr_best.pt` to use it at inference time.
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user