add 315k YOLOv8 model

.gitattributes

@@ -33,3 +33,29 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+examples/ filter=lfs diff=lfs merge=lfs -text
+weights/ filter=lfs diff=lfs merge=lfs -text
+weights/mbari_315k_yolov8.pt filter=lfs diff=lfs merge=lfs -text
+examples/soft_coral.png filter=lfs diff=lfs merge=lfs -text
+examples/jelly_2.png filter=lfs diff=lfs merge=lfs -text
+examples/jelly.png filter=lfs diff=lfs merge=lfs -text
+examples/puff.png filter=lfs diff=lfs merge=lfs -text
+examples/red_fish_2.png filter=lfs diff=lfs merge=lfs -text
+examples/scene_2.png filter=lfs diff=lfs merge=lfs -text
+examples/fish_2.png filter=lfs diff=lfs merge=lfs -text
+examples/fish.png filter=lfs diff=lfs merge=lfs -text
+examples/scene_5.png filter=lfs diff=lfs merge=lfs -text
+examples/scene.png filter=lfs diff=lfs merge=lfs -text
+examples/fish_4.png filter=lfs diff=lfs merge=lfs -text
+examples/fish_5.png filter=lfs diff=lfs merge=lfs -text
+examples/flat_fish.png filter=lfs diff=lfs merge=lfs -text
+examples/red_fish.png filter=lfs diff=lfs merge=lfs -text
+examples/starfish_2.png filter=lfs diff=lfs merge=lfs -text
+examples/scene_4.png filter=lfs diff=lfs merge=lfs -text
+examples/scene_6.png filter=lfs diff=lfs merge=lfs -text
+examples/starfish.png filter=lfs diff=lfs merge=lfs -text
+examples/crab.png filter=lfs diff=lfs merge=lfs -text
+examples/fish_3.png filter=lfs diff=lfs merge=lfs -text
+examples/flat_red_fish.png filter=lfs diff=lfs merge=lfs -text
+examples/jelly_3.png filter=lfs diff=lfs merge=lfs -text
+examples/scene_3.png filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,8 +1,8 @@
 ---
 title: MBARI 315k
-emoji: 🐠
-colorFrom: gray
-colorTo: red
+emoji: 🐙
+colorFrom: blue
+colorTo: gray
 sdk: gradio
 sdk_version: 5.25.1
 app_file: app.py
@@ -10,5 +10,3 @@ pinned: false
 license: mit
 short_description: MBARI 315k models trained on FathomNet and VARS data
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,34 @@
+from pathlib import Path
+
+import gradio as gr
+from ultralytics import YOLO
+
+# Load models
+MODEL_PATH = "weights/mbari_315k_yolov8.pt"
+MODEL = YOLO(MODEL_PATH)
+
+PREDICT_KWARGS = {
+    "conf": 0.15,
+}
+
+# Get example images
+EXAMPLES_DIR = Path("examples")
+EXAMPLES = list(EXAMPLES_DIR.glob("*.png")) if EXAMPLES_DIR.exists() else []
+
+
+def detect_objects(image):
+    results = MODEL.predict(image, **PREDICT_KWARGS)
+    return results[0].plot()
+
+
+# Gradio interface with gr.Interface instead of gr.Blocks
+demo = gr.Interface(
+    fn=detect_objects,
+    inputs=gr.Image(type="numpy"),
+    outputs=gr.Image(type="numpy"),
+    title="MBARI 315k",
+    examples=EXAMPLES if EXAMPLES else None,
+    cache_examples=True,
+)
+
+demo.queue().launch()

requirements.txt

@@ -0,0 +1 @@
+ultralytics==8.3.108

weights/mbari_315k_yolov8.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:16d2d8f066501b58d35125f172a34fa1b51add9eaab25a8ecd6de92932ef775d
+size 137709813