Initial commit with YOLOv5 Sunspot Hunter app

README.md

@@ -1,6 +1,6 @@
 ---
 title: Yolov5 Sunspot Hunter
-emoji: 🌖
+emoji: 🌟
 colorFrom: green
 colorTo: blue
 sdk: gradio
@@ -10,4 +10,16 @@ pinned: false
 license: ecl-2.0
 ---
 
+# YOLOv5 Sunspot Hunter
+
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+## Desenvolvedor
+
+Desenvolvido por Ramon Mayor Martins (2023)
+
+- Email: [rmayormartins@gmail.com](mailto:rmayormartins@gmail.com)
+- Homepage: [https://rmayormartins.github.io/](https://rmayormartins.github.io/)
+- Twitter: [@rmayormartins](https://twitter.com/rmayormartins)
+- GitHub: [https://github.com/rmayormartins](https://github.com/rmayormartins)
+

app.py

@@ -0,0 +1,47 @@
+import gradio as gr
+import numpy as np
+from PIL import Image
+import torch
+import matplotlib.pyplot as plt
+
+model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')
+
+
+def detect(img):
+    
+    img_arr = np.array(img)
+    
+    results = model(img_arr)
+
+    
+    fig, ax = plt.subplots()
+    ax.imshow(img_arr)
+
+    
+    for *xyxy, conf, cls in results.xyxy[0].cpu().numpy():
+        x1, y1, x2, y2 = map(int, xyxy)
+        label = model.names[int(cls)]  
+        ax.add_patch(plt.Rectangle((x1, y1), x2-x1, y2-y1, fill=False, color='red', linewidth=2))
+        ax.text(x1, y1, f'{label} {conf:.2f}', color='white', fontsize=8, bbox={'facecolor': 'red', 'alpha': 0.5})
+
+    plt.axis('off')  
+
+    
+    fig.canvas.draw()
+    pil_img = Image.fromarray(np.array(fig.canvas.renderer._renderer))
+    plt.close(fig)  
+
+    return pil_img
+
+#Gradio
+iface = gr.Interface(
+    fn=detect,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Image(type="pil"),
+    title="YOLOv5 Sun Spot Hunter",
+    description="Object detector (solar spot/sunspot hunter) trained using YOLOv5 and labeled in Makesense.ai",
+    examples=[["example1.jpg"]]  
+)
+
+
+iface.launch(debug=True)

best.pt

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

requirements.txt

@@ -0,0 +1,6 @@
+gradio
+torch
+numpy
+Pillow
+matplotlib
+