jwlarocque/yolov8n-freeclimbs-detect-2
By jwlarocque · ↓ 0 · ♥ 2 · Updated 2024-02-07
What's in this repo
This page lists the files in jwlarocque/yolov8n-freeclimbs-detect-2, including yolov8n-freeclimbs-detect-2-fp16.onnx, yolov8n-freeclimbs-detect-2-fp32.onnx.
- 2 ONNX models (.onnx files)
- 6 files, including 2 .onnx, 1 .md, 1 .pt
- Last updated 2024-02-07
ONNX files
| File | Size | Download |
|---|---|---|
| yolov8n-freeclimbs-detect-2-fp16.onnx | — | Download |
| yolov8n-freeclimbs-detect-2-fp32.onnx | — | Download |
How to use
These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.
README
This model is a version of Yolo v8 nano fine-tuned on the freeclimbs v2 dataset to detect climbing holds, particularly holds on home climbing and "spray" walls. (The dataset is not currently available but I plan to release it in the future.)
It expects a 2560x2560 image (if using the ultralytics library as shown below, it will handle this) and detects a single class - climbing holds.
Usage
from ultralytics import YOLO
model = YOLO("yolov8n-freeclimbs-detect-2.pt")
results = model(
["climbing-wall.jpg"],
imgsz=2560,
max_det=2000)
Performance
| Precision | 0.961 |
| Recall | 0.942 |
| mAP50 | 0.988 |
| mAP50-95 | 0.889 |
(on freeclimbs v2 test set)
License
Copyright (c) 2024 John LaRocque
See LICENSE for license (AGPL 3). Note that an earlier version of this repository erroneously included an MIT license - since this model was fine-tuned from a model licensed under the AGPL 3, which is incompatible with other licenses, I am not actually able to offer that license.
Model info
Created: 2024-02-04
Updated: 2024-02-07
View on Hugging FaceFiles (6)
Related models
More from jwlarocque
No other models by this author.