Add Everything

.gitignore

@@ -0,0 +1 @@
+**/.DS_Store

Content/best.pt

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

Content/configurations.md

@@ -0,0 +1,7 @@
+model.conf = 0.25  # NMS confidence threshold
+      iou = 0.45  # NMS IoU threshold
+      agnostic = False  # NMS class-agnostic
+      multi_label = False  # NMS multiple labels per box
+      classes = None  # (optional list) filter by class, i.e. = [0, 15, 16] for COCO persons, cats and dogs
+      max_det = 1000  # maximum number of detections per image
+      amp = False  # Automatic Mixed Precision (AMP) inference

Content/data.yaml

@@ -0,0 +1,8 @@
+names:
+- Airplane
+- Ship
+- Tree
+- Vehicle
+nc: 4
+train: /content/datasets/Palmyra-Classification-1/train/images
+val: /content/datasets/Palmyra-Classification-1/valid/images

app.py

@@ -0,0 +1,41 @@
+import gradio as gr
+import os
+import torch
+from PIL import Image
+
+                           
+#subprocess.run(["mv","content/custom_data.yaml","./yolov5/data"])         
+
+
+def load_model():
+ '''
+ Loading hub model & setting the preferences for the model  
+ '''
+ model = torch.hub.load('ultralytics/yolov5', 'custom', path='Content/best.pt')
+ model.conf = 0.38 
+ model.dnn=True
+ model.agnostic=True
+ return model
+
+model=load_model()
+#, force_reload=True
+def detect(inp):
+ #g = (size / max(inp.size))  #gain
+ #im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS)  # resize 
+ results = model(inp,size=640)  # inference
+ print(results)
+ results.render()  # updates results.imgs with boxes and labels
+ return Image.fromarray(results.imgs[0])
+ 
+
+inp = gr.inputs.Image(type="pil", label="Original Image")
+output = gr.outputs.Image(type="pil", label="Output Image")
+
+
+io=gr.Interface(fn=detect, inputs=inp, outputs=output, title='A3 Satellite Classification',theme='peach')
+io.launch(debug=True,share=False)
+  
+#examples=['Content/4.jpg','Content/10.jpg','Content/18.jpg']
+
+                                             
+                                                                                                                                       

packages.txt

@@ -0,0 +1 @@
+libgl1-mesa-glx

requirements.txt

@@ -0,0 +1,20 @@
+gradio
+
+# Base ----------------------------------------
+matplotlib>=3.2.2
+numpy>=1.18.5
+opencv-python>=4.1.2
+Pillow>=7.1.2
+PyYAML>=5.3.1
+requests>=2.23.0
+scipy>=1.4.1
+torch>=1.7.0
+torchvision>=0.8.1
+tqdm>=4.41.0
+
+
+
+# Plotting ------------------------------------
+pandas>=1.1.4
+seaborn>=0.11.0
+