Ocean Yolo

Usage

Requires YOLOv5.

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()

sonar_detector/best.pt

pt · 8.10 MB · SHA-256 7d651083a1ec…a6c2 · Hugging Face

--- 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()