init

.gitignore

@@ -0,0 +1 @@
+.venv

app.py

@@ -0,0 +1,27 @@
+import gradio as gr
+from ultralytics import YOLO
+import numpy as np
+
+# Load YOLO model
+model = YOLO('./best.pt')
+
+def process_image(input_image):
+    if input_image is not None:
+        results = model(input_image)
+
+        for r in results:
+            im_array = r.plot()
+            im_array = im_array.astype(np.uint8)
+            return im_array
+
+# Create Gradio Interface
+iface = gr.Interface(
+    fn=process_image,
+    inputs=gr.Image(),
+    outputs=gr.Image(),  # Specify output as Gradio Image
+    title="YOLOv8-obb aerial detection",
+    description="YOLOv8-obb trained on DOTAv1.5",
+    live=True)
+
+# Launch the Gradio interface
+iface.launch()

best.pt

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

requirements.txt

@@ -0,0 +1,119 @@
+aiofiles==23.2.1
+altair==5.2.0
+annotated-types==0.6.0
+anyio==4.3.0
+asttokens==2.4.1
+attrs==23.2.0
+beautifulsoup4==4.12.3
+certifi==2023.7.22
+chardet==4.0.0
+charset-normalizer==3.3.2
+clearml==1.14.3
+click==8.1.7
+colorama==0.4.6
+comm==0.2.1
+contourpy==1.2.0
+cycler==0.10.0
+Cython==3.0.8
+debugpy==1.8.1
+decorator==5.1.1
+defusedxml==0.7.1
+executing==2.0.1
+fastapi==0.109.2
+ffmpy==0.3.2
+filelock==3.13.1
+fonttools==4.48.1
+fsspec==2024.2.0
+furl==2.1.3
+gdown==5.1.0
+gradio==4.19.1
+gradio_client==0.10.0
+h11==0.14.0
+httpcore==1.0.3
+httpx==0.26.0
+huggingface-hub==0.20.3
+idna==2.10
+importlib-resources==6.1.1
+ipykernel==6.29.2
+ipython==8.21.0
+jedi==0.19.1
+Jinja2==3.1.3
+jsonschema==4.21.1
+jsonschema-specifications==2023.12.1
+jupyter_client==8.6.0
+jupyter_core==5.7.1
+kiwisolver==1.4.5
+lapx==0.5.5
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+matplotlib==3.8.2
+matplotlib-inline==0.1.6
+mdurl==0.1.2
+mpmath==1.3.0
+nest-asyncio==1.6.0
+networkx==3.2.1
+numpy==1.26.4
+opencv-python==4.9.0.80
+opencv-python-headless==4.8.0.74
+orderedmultidict==1.0.1
+orjson==3.9.14
+packaging==23.2
+pandas==2.2.0
+parso==0.8.3
+pathlib2==2.3.7.post1
+pexpect==4.9.0
+pillow==10.2.0
+platformdirs==4.2.0
+prompt-toolkit==3.0.43
+psutil==5.9.8
+ptyprocess==0.7.0
+pure-eval==0.2.2
+py-cpuinfo==9.0.0
+pydantic==2.6.1
+pydantic_core==2.16.2
+pydub==0.25.1
+Pygments==2.17.2
+PyJWT==2.8.0
+pyparsing==3.1.1
+PySocks==1.7.1
+python-dateutil==2.8.2
+python-dotenv==1.0.1
+python-magic==0.4.27
+python-multipart==0.0.9
+pytz==2024.1
+PyYAML==6.0.1
+pyzmq==25.1.2
+referencing==0.33.0
+requests==2.31.0
+requests-toolbelt==1.0.0
+rich==13.7.0
+roboflow==1.1.19
+rpds-py==0.18.0
+ruff==0.2.2
+scipy==1.12.0
+seaborn==0.13.2
+semantic-version==2.10.0
+shellingham==1.5.4
+six==1.16.0
+sniffio==1.3.0
+soupsieve==2.5
+stack-data==0.6.3
+starlette==0.36.3
+supervision==0.18.0
+sympy==1.12
+thop==0.1.1.post2209072238
+tomlkit==0.12.0
+toolz==0.12.1
+torch==2.2.0
+torchvision==0.17.0
+tornado==6.4
+tqdm==4.66.2
+traitlets==5.14.1
+typer==0.9.0
+typing_extensions==4.9.0
+tzdata==2024.1
+ultralytics==8.1.11
+urllib3==2.2.0
+uvicorn==0.27.1
+wcwidth==0.2.13
+websockets==11.0.3