| Class | AP |
|---|---|
| cement_bags | 77.2% |
| red bricks | 64.8% |
| helmet | 49.2% |
| safety_vest | 47.4% |
| steel bar | 39.4% |
| no_helmet | 39.4% |
| concrete_blocks | 38.7% |
| no_safety_vest | 21.8% |
1import torch
2ckpt = torch.load("astra_vision_v1_best.pth", map_location="cpu")
3model.load_state_dict(ckpt["model"])