Ocean-router

zjunlp/Ocean-router

Ocean-router is a image classification model from zjunlp. Use it when you need a label for an image. It is set up for ultralytics. The card lists the license as apache-2.0.

Two-stage lightweight classifiers used to dynamically route marine images to specialized detectors based on image modality and content.

Downloads · 30 days

3% of all-time downloads

All-time downloads

169

Public

Repo size

11.6 MB

Likes

Public

Hugging Face

Repo makeup

Click a slice to open those files.

.pt11.6 MB · 100%

At a glance

Task: Image Classification
Library: ultralytics
License: apache-2.0
Access: Public
Created: May 10, 2026
Updated: May 10, 2026
SHA: 510250ca

Model card

From the Hugging Face model README

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

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

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.

Task: Image Classification
Library: ultralytics
License: apache-2.0
Languages: en
Created: May 10, 2026
Updated: May 10, 2026