Views
No views yet
| File | Task | Classes | Architecture |
|---|---|---|---|
sonar_detector/best.pt | Sonar target detection | 15 sonar categories | YOLOv5 |
fish_detector/best.pt | Fish species/object detection | Multiple fish classes | YOLOv5 |
coral_detector/best.pt | Coral species/object detection | Multiple coral classes | YOLOv5 |
1import torch
2# Load detector (e.g., Fish)
3model = torch.hub.load("ultralytics/yolov5", "custom", path="fish_detector/best.pt", force_reload=True)
4# Inference
5results = model("input.jpg", conf=0.5, iou=0.45)
6results.print()