updated models. Fix colors

app.py

@@ -2,6 +2,7 @@ import os
 import gradio as gr
 from ultralytics import YOLO
 import numpy as np
+import cv2
 
 model_options = ["yolo-8n-dota.pt", "yolo-8s-dota.pt", "yolo-8m-dota.pt"]
 model_names = ["Nano", "Small", "Medium"]
@@ -18,6 +19,12 @@ def process_image(input_image, model_name, conf):
     if conf is None:
         conf = 0.6
 
+    orig_height, orig_width, _ = input_image.shape
+    print(input_image.shape)
+
+    input_image = cv2.cvtColor(input_image, cv2.COLOR_RGB2BGR)
+    cv2.imwrite('./res.jpg', input_image)
+
     model_index = model_names.index(model_name)
     model = models[model_index]
     
@@ -28,6 +35,7 @@ def process_image(input_image, model_name, conf):
     for r in results:
         im_array = r.plot()
         im_array = im_array.astype(np.uint8)
+        im_array = cv2.cvtColor(im_array, cv2.COLOR_BGR2RGB)
 
         if len(r.obb.cls) == 0:  # If no objects are detected
             return None, "No objects detected."

yolo-8m-dota.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f6d0e873f8847ea3e222f4ae8a1d3f50266d9c0c0c1f636b6bb2a3c5a7291976
-size 53237546
+oid sha256:19906b037246aa964742ec8560fc8c3e48b479542bcd37591e7a01a108d3ac09
+size 53237418

yolo-8n-dota.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4ee44108137e10e13a377e2a75175af1476c355e04217cc38e3e8e2f4cb6fd7c
+oid sha256:1918f8430c5918a7f866c9be5d6099bd3d2601f1d290fa4fe08096658b75bc84
 size 6465538

yolo-8s-dota.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4395070f75af8464e3c4d7e7d83eda61d19d6cd5ea6c62ca1898844a8e0ad54c
-size 23169282
+oid sha256:999b8ff11aaffeffbb7e9a559802ed25f9c8d24c94d407b0adead1007175347e
+size 23168706