Update README.md

README.md

@@ -1,25 +1,35 @@
----
-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 classification | YOLOv5 | Image → `[fish_prob, coral_prob]` |
-## Usage
-### 1. Sonar/Biological Router (YOLOv11)
-```python
-from ultralytics import YOLO
-router = YOLO("cls_bio_sonar/best.pt")
-results = router.predict("input.jpg")
+---
+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.