zhemaxiya commited on
Commit
226c809
·
verified ·
1 Parent(s): cbf41bb

Upload README.md

Browse files
Files changed (1) hide show
  1. README.md +29 -0
README.md ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ license: apache-2.0
4
+ library_name: yolov5
5
+ tags:
6
+ - yolov5
7
+ - object-detection
8
+ - fish-detection
9
+ - coral-detection
10
+ - sonar-detection
11
+ - oceangpt-x
12
+ ---
13
+ # Ocean-yolo: Marine Object Detectors
14
+ Specialized YOLOv5 detection models for fine-grained localization and classification of marine targets. Invoked after the routing stage.
15
+ ## Model Weights
16
+ | File | Task | Classes | Architecture |
17
+ |------|------|---------|--------------|
18
+ | `sonar_detector/best.pt` | Sonar target detection | 15 sonar categories | YOLOv5 |
19
+ | `fish_detector/best.pt` | Fish species/object detection | Multiple fish classes | YOLOv5 |
20
+ | `coral_detector/best.pt` | Coral species/object detection | Multiple coral classes | YOLOv5 |
21
+ ## Usage
22
+ Requires [YOLOv5](https://github.com/ultralytics/yolov5).
23
+ ```python
24
+ import torch
25
+ # Load detector (e.g., Fish)
26
+ model = torch.hub.load("ultralytics/yolov5", "custom", path="fish_detector/best.pt", force_reload=True)
27
+ # Inference
28
+ results = model("input.jpg", conf=0.5, iou=0.45)
29
+ results.print()