add gradio app

.gitattributes

@@ -33,3 +33,6 @@ 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
+weights/megafishdetector_v0_yolov5l_640p.pt filter=lfs diff=lfs merge=lfs -text
+weights/megafishdetector_v0_yolov5m_1280p.pt filter=lfs diff=lfs merge=lfs -text
+weights/megafishdetector_v0_yolov5s_640p.pt filter=lfs diff=lfs merge=lfs -text

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2023 WHOI's Autonomous Robotics and Perception Lab
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

README.md

@@ -1,6 +1,6 @@
 ---
-title: Megafishdetector
-emoji: 📚
+title: megafishdetector
+emoji: 🐟
 colorFrom: green
 colorTo: purple
 sdk: gradio
@@ -10,5 +10,3 @@ pinned: false
 license: mit
 short_description: WARPLab detector for generic "fish"
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,39 @@
+import gradio as gr
+import yolov5
+
+# Load models
+model_paths = {
+    "YOLOv5 Large 640p": "weights/megafishdetector_v0_yolov5l_640p.pt",
+    "YOLOv5 Medium 1280p": "weights/megafishdetector_v0_yolov5m_1280p.pt",
+    "YOLOv5 Small 640p": "weights/megafishdetector_v0_yolov5s_640p.pt",
+}
+
+models = {name: yolov5.load(path) for name, path in model_paths.items()}
+
+
+def detect_objects(image, model_name):
+    model = models[model_name]
+    results = model(image)
+    return results.render()[0]
+
+
+# Gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown("# megafishdetector")
+
+    with gr.Row():
+        with gr.Column():
+            image_input = gr.Image(type="numpy")
+            model_selector = gr.Dropdown(
+                choices=list(model_paths.keys()), label="Select Model"
+            )
+            submit_button = gr.Button("Submit")
+
+        with gr.Column():
+            image_output = gr.Image(type="numpy")
+
+    submit_button.click(
+        fn=detect_objects, inputs=[image_input, model_selector], outputs=image_output
+    )
+
+demo.launch()

requirements.txt

@@ -0,0 +1,2 @@
+yolov5==7.0.14
+ultralytics==8.3.78

weights/README.md

@@ -0,0 +1,17 @@
+# Models
+
+## MegaFishDetector v0 (single class):
+[Link to all weights](https://drive.google.com/drive/folders/14fA7djYvxchfdqgWnCJABV5JQETjVe0I?usp=sharing)
+
+Models you'll find:
+- megafishdetector_v0_yolov5s_640p.pt (trained 600 epochs)
+- megafishdetector_v0_yolov5m_1280p.pt (trained 150 epochs)
+- megafishdetector_v0_yolov5l_640p.pt (trained 450 epochs)
+
+Data trained on:
+- [AIMs Ozfish](https://github.com/open-AIMS/ozfish) 
+- [FathomNet (Gnathostomata phyla only)](https://www.fathomnet.org/)
+- [VIAME FishTrack (50% subsampled)](https://viame.kitware.com/#/collection/62afcb66dddafb68c8442126)
+- [NOAA Puget Sound Nearshore Fish (2017-2018)](https://lila.science/datasets/noaa-puget-sound-nearshore-fish)
+- [DeepFish](https://alzayats.github.io/DeepFish/)
+- [NOAA Labelled Fishes in the Wild](https://www.st.nmfs.noaa.gov/aiasi/DataSets.html)

weights/megafishdetector_v0_yolov5l_640p.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:47117e50f9625bc83ea1d408f86145c76dc1faa6cf9c63cbc25c2d88c8cf1377
+size 370061843

weights/megafishdetector_v0_yolov5m_1280p.pt

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

weights/megafishdetector_v0_yolov5s_640p.pt

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