Views
No views yet
| ID | Class |
|---|---|
| 0 | scissors |
| 1 | knife |
| 2 | cutter |
| 3 | syringe |
.ptweights/best.pt| Revision | Training epochs | Best mAP50 | Best mAP50-95 |
|---|---|---|---|
hazard-objects-v1-full-v1 | 100 | 0.8681 | 0.6555 |
hazard-objects-v2-finetune-v1 | 11 | 0.7721 | 0.5359 |
hazard-objects-v2-finetune-v1-extend20 | 20 | 0.8260 | 0.6409 |
hazard-objects-v2-finetune-v2 | 30 | 0.8322 | 0.6388 |
hazard-objects-v2-finetune-v2-extend20 | 8 | 0.8211 | 0.6202 |
hazard-objects-v2-finetune-v3 | 40 | 0.8381 | 0.6113 |
hazard-objects-c270-finetune-v1 | 20 | 0.8348 | 0.6382 |
1from huggingface_hub import hf_hub_download
2from ultralytics import YOLO
3
4weight_path = hf_hub_download(
5 repo_id="chocochip119/wardy-m05-hazard-detector",
6 filename="weights/best.pt",
7 revision="main",
8)
9
10model = YOLO(weight_path)
11results = model.predict(
12 source="example.jpg",
13 imgsz=640,
14 conf=0.25,
15 save=True,
16)