---
language: en
license: apache-2.0
library_name: yolov5
tags:
- yolov5
- object-detection
- fish-detection
- coral-detection
- sonar-detection
- oceangpt-x
---
# Ocean-yolo: Marine Object Detectors
Specialized YOLOv5 detection models for fine-grained localization and classification of marine targets. Invoked after the routing stage.
## Model Weights
| 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 |
## Usage
Requires [YOLOv5](https://github.com/ultralytics/yolov5).
```python
import torch
# Load detector (e.g., Fish)
model = torch.hub.load("ultralytics/yolov5", "custom", path="fish_detector/best.pt", force_reload=True)
# Inference
results = model("input.jpg", conf=0.5, iou=0.45)
results.print()