PP-YOLOE
PP-YOLOE is an anchor-free object detector with detection training and class-head resizing.
- 任务
- detection
- 尺寸
- s, m, l, x at 640 px
- 安装
pip install libreyolo- 支持层级
- 已支持,自 v1.6.0 起。起支撑作用的可训练家族:在 CI 里保持绿色,功能视情况落地。
- 许可
- 代码采用 MIT,权重采用 No separate artifact grant recorded。商用
Install
pip install "libreyolo"Predict
from libreyolo import LibreYOLO, SAMPLE_IMAGE model = LibreYOLO("LibrePPYOLOEs.pt", device="cpu")result = model(SAMPLE_IMAGE)print(result.boxes)Released weights download from the source CDN with a pinned SHA-256 check. The constructor also accepts a local upstream checkpoint. The four sizes are s, m, l and x.
Train
Training uses ATSS assignment before switching to task-aligned assignment. static_assigner_epochs defaults to 30% of the requested epochs. A new class count rebuilds the class-prediction layers. AMP defaults to False.
Dataset setup describes the required training data.
from libreyolo import LibreYOLO model = LibreYOLO("LibrePPYOLOEs.pt", device="cpu")# coco8 is a small YOLO-format detection dataset that downloads automatically.model.train(data="coco8.yaml", epochs=1, device="cpu", workers=0)Validate
from libreyolo import LibreYOLO model = LibreYOLO("LibrePPYOLOEs.pt", device="cpu")metrics = model.val(data="coco8.yaml", workers=0)print(metrics)Use the dataset format for this task. Validation explains the dataset requirements and returned metrics.
Export
| 任务 | ONNX | TorchScript | ExecuTorch | TensorRT | OpenVINO | Paddle | MNN | RKNN | ncnn | TFLite | CoreML | Core AI |
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Detection | Detection to ONNX:支持 | Detection to TorchScript:支持 | Detection to ExecuTorch:支持 | Detection to TensorRT:支持 | Detection to OpenVINO:支持 | Detection to Paddle:不支持 | Detection to MNN:不支持 | Detection to RKNN:不支持 | Detection to ncnn:支持 | Detection to TFLite:不支持 | Detection to CoreML:不支持 | Detection to Core AI:不支持 |
from libreyolo import LibreYOLO model = LibreYOLO("LibrePPYOLOEs.pt", device="cpu")model.export(format="onnx")Export setup lists format dependencies and loading exported artifacts.
Licensing
请检查你所下载的具体权重在 Hugging Face 仓库中的许可。LibreYOLO 组织里的每个检查点都附有许可,同一家族内也不一定相同。该仓库是权威来源;以下摘要说明本页上次验证时适用的情况。
这里只说明涉及的许可证,不构成法律意见。如果答案对商用很重要,请自行阅读许可证并咨询法律顾问。
- 原始工作
- PP-YOLOE, PaddlePaddle
- 上游许可
- Apache-2.0
- LibreYOLO 代码
- MIT
- 权重
- 采用 No separate artifact grant recorded 许可,由作者分发。LibreYOLO 不托管或镜像这些权重。
- 解读
- The upstream source is Apache-2.0. Pretrained artifacts download from the source CDN; a separate checkpoint license grant has not been recorded.