| --- |
| language: en |
| license: apache-2.0 |
| library_name: ultralytics |
| tags: |
| - yolov11 |
| - yolov5 |
| - image-classification |
| - routing |
| - marine-images |
| - oceangpt-x |
| --- |
| # Ocean-router: Marine Image Routing Classifiers |
| Two-stage lightweight classifiers used to dynamically route marine images to specialized detectors based on image modality and content. |
| ## Model Files & Tasks |
| | File | Task | Architecture | Input/Output | |
| |------|------|--------------|--------------| |
| | `cls_bio_sonar/best.pt` | Sonar vs. Biological routing | YOLOv11-cls | Image → `[sonar_prob, bio_prob]` | |
| | `fish_coral_cls/best.pt` | Fish vs. Coral routing | YOLOv5 | Image → `[fish_prob, coral_prob]` | |
| ## Usage |
| ### 1. Sonar/Biological Router |
| ```python |
| from ultralytics import YOLO |
| router = YOLO("cls_bio_sonar/best.pt") |
| results = router.predict("input.jpg") |
| ``` |
| ### 2. Fish/Coral Classifier (YOLOv5) |
| Requires the official YOLOv5 repo |
| ```python |
| import torch |
| model = torch.hub.load("ultralytics/yolov5", "custom", path="fish_coral_cls/best.pt", force_reload=True) |
| results = model("input.jpg") |
| ``` |
| cls_bio_sonar decides if input is sonar or biological. |
| If biological, fish_coral_cls routes to the appropriate species detector. |
