Duplicate from hlydecker/MegaDetector_v5

Co-authored-by: Henry Lydecker <[email protected]>

.gitattributes

@@ -0,0 +1,28 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zstandard filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text
+data/truck.png filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,14 @@
+---
+title: MegaDetector V5
+emoji: 📊
+colorFrom: purple
+colorTo: green
+sdk: gradio
+sdk_version: 3.0.19
+app_file: app.py
+pinned: false
+license: mit
+duplicated_from: hlydecker/MegaDetector_v5
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,36 @@
+# MegaDetector v5 Demo
+import gradio as gr
+import torch
+import torchvision
+import numpy as np
+from PIL import Image
+
+# Load MegaDetector v5a model
+# TODO: Allow user selectable model?
+# models = ["model_weights/md_v5a.0.0.pt","model_weights/md_v5b.0.0.pt"]
+model = torch.hub.load('ultralytics/yolov5:v6.2', 'custom', "model_weights/md_v5a.0.0.pt")
+
+def yolo(im, size=640):
+    g = (size / max(im.size))  # gain
+    im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS)  # resize
+    
+    model = torch.hub.load('ultralytics/yolov5:v6.2', 'custom', "model_weights/md_v5a.0.0.pt")
+    
+    results = model(im)  # inference
+    results.render()  # updates results.imgs with boxes and labels
+    return Image.fromarray(results.imgs[0])
+
+#image = gr.inputs.Image(type="pil", label="Input Image")
+#chosen_model = gr.inputs.Dropdown(choices = models, value = "model_weights/md_v5a.0.0.pt",type = "value", label="Model Weight")
+#size = 640
+
+#inputs = [image, chosen_model, size]
+inputs = gr.inputs.Image(type="pil", label="Input Image")
+outputs = gr.outputs.Image(type="pil", label="Output Image")
+
+title = "MegaDetector v5"
+description = "Detect and identify animals, people and vehicles in camera trap images using MegaDetector v5a"
+article = "<p style='text-align: center'>This app makes predictions using a YOLOv5x6 model that was trained to detect animals, humans, and vehicles in camera trap images; find out more about the project on <a href='https://github.com/microsoft/CameraTraps'>GitHub</a>. This app was built by Henry Lydecker but really depends on code and models developed by <a href='http://ecologize.org/'>Ecologize</a> and <a href='http://aka.ms/aiforearth'>Microsoft AI for Earth</a>. Find out more about the YOLO model from the original creator, <a href='https://pjreddie.com/darknet/yolo/'>Joseph Redmon</a>. YOLOv5 is a family of compound-scaled object detection models trained on the COCO dataset and developed by Ultralytics, and includes simple functionality for Test Time Augmentation (TTA), model ensembling, hyperparameter evolution, and export to ONNX, CoreML and TFLite. <a href='https://github.com/ultralytics/yolov5'>Source code</a> | <a href='https://pytorch.org/hub/ultralytics_yolov5'>PyTorch Hub</a></p>"
+
+examples = [['data/Macropod.jpg'], ['data/koala2.jpg'],['data/cat.jpg'],['data/BrushtailPossum.jpg']]
+gr.Interface(yolo, inputs, outputs, title=title, description=description, article=article, examples=examples, theme="huggingface").launch(enable_queue=True)

data/README.md

@@ -0,0 +1,3 @@
+# Data
+
+Example images go here.

model_weights/README.md

@@ -0,0 +1,3 @@
+# Model Weights folder
+
+Model weights should go here, if they aren't available on some sort of hub somewhere.

model_weights/md_v5a.0.0.pt

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

model_weights/md_v5b.0.0.pt

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

requirements.txt

@@ -0,0 +1,11 @@
+# pip install -r requirements.txt
+
+# Requirements for are you wearing a mask
+# A YOLOv5 powered face mask decector Gradio application
+
+gradio
+torch
+torchvision
+numpy
+opencv-python
+seaborn