Views
No views yet
| Metric | Value |
|---|---|
| mAP@50 | 0.7358 |
| mAP@50-95 | 0.5381 |
| Precision | 0.7482 |
| Recall | 0.6811 |
| Train Images | 2285 |
| Val Images | 404 |
| ID | Class |
|---|---|
| 0 | bed |
| 1 | sofa |
| 2 | chair |
| 3 | table |
| 4 | lamp |
| 5 | tv |
| 6 | laptop |
| 7 | wardrobe |
| 8 | window |
| 9 | door |
| 10 | potted plant |
| 11 | photo frame |
1from ultralytics import YOLO
2
3model = YOLO('best.pt')
4results = model('room_photo.jpg', conf=0.25)
5
6CLASS_NAMES = ['bed', 'sofa', 'chair', 'table', 'lamp', 'tv', 'laptop', 'wardrobe', 'window', 'door', 'potted plant', 'photo frame']
7for box in results[0].boxes:
8 name = CLASS_NAMES[int(box.cls[0])]
9 conf = float(box.conf[0])
10 print(f'{name}: {conf:.2f}')