VlaTal commited on
Commit
a190fc6
1 Parent(s): 9bbd29b
Files changed (4) hide show
  1. .gitignore +1 -0
  2. app.py +27 -0
  3. best.pt +3 -0
  4. requirements.txt +119 -0
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ .venv
app.py ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from ultralytics import YOLO
3
+ import numpy as np
4
+
5
+ # Load YOLO model
6
+ model = YOLO('./best.pt')
7
+
8
+ def process_image(input_image):
9
+ if input_image is not None:
10
+ results = model(input_image)
11
+
12
+ for r in results:
13
+ im_array = r.plot()
14
+ im_array = im_array.astype(np.uint8)
15
+ return im_array
16
+
17
+ # Create Gradio Interface
18
+ iface = gr.Interface(
19
+ fn=process_image,
20
+ inputs=gr.Image(),
21
+ outputs=gr.Image(), # Specify output as Gradio Image
22
+ title="YOLOv8-obb aerial detection",
23
+ description="YOLOv8-obb trained on DOTAv1.5",
24
+ live=True)
25
+
26
+ # Launch the Gradio interface
27
+ iface.launch()
best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3cb204d0119354b16c00bb51bb3bdaa8604eca9d56e5af7d6207187e6684b169
3
+ size 53365226
requirements.txt ADDED
@@ -0,0 +1,119 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ aiofiles==23.2.1
2
+ altair==5.2.0
3
+ annotated-types==0.6.0
4
+ anyio==4.3.0
5
+ asttokens==2.4.1
6
+ attrs==23.2.0
7
+ beautifulsoup4==4.12.3
8
+ certifi==2023.7.22
9
+ chardet==4.0.0
10
+ charset-normalizer==3.3.2
11
+ clearml==1.14.3
12
+ click==8.1.7
13
+ colorama==0.4.6
14
+ comm==0.2.1
15
+ contourpy==1.2.0
16
+ cycler==0.10.0
17
+ Cython==3.0.8
18
+ debugpy==1.8.1
19
+ decorator==5.1.1
20
+ defusedxml==0.7.1
21
+ executing==2.0.1
22
+ fastapi==0.109.2
23
+ ffmpy==0.3.2
24
+ filelock==3.13.1
25
+ fonttools==4.48.1
26
+ fsspec==2024.2.0
27
+ furl==2.1.3
28
+ gdown==5.1.0
29
+ gradio==4.19.1
30
+ gradio_client==0.10.0
31
+ h11==0.14.0
32
+ httpcore==1.0.3
33
+ httpx==0.26.0
34
+ huggingface-hub==0.20.3
35
+ idna==2.10
36
+ importlib-resources==6.1.1
37
+ ipykernel==6.29.2
38
+ ipython==8.21.0
39
+ jedi==0.19.1
40
+ Jinja2==3.1.3
41
+ jsonschema==4.21.1
42
+ jsonschema-specifications==2023.12.1
43
+ jupyter_client==8.6.0
44
+ jupyter_core==5.7.1
45
+ kiwisolver==1.4.5
46
+ lapx==0.5.5
47
+ markdown-it-py==3.0.0
48
+ MarkupSafe==2.1.5
49
+ matplotlib==3.8.2
50
+ matplotlib-inline==0.1.6
51
+ mdurl==0.1.2
52
+ mpmath==1.3.0
53
+ nest-asyncio==1.6.0
54
+ networkx==3.2.1
55
+ numpy==1.26.4
56
+ opencv-python==4.9.0.80
57
+ opencv-python-headless==4.8.0.74
58
+ orderedmultidict==1.0.1
59
+ orjson==3.9.14
60
+ packaging==23.2
61
+ pandas==2.2.0
62
+ parso==0.8.3
63
+ pathlib2==2.3.7.post1
64
+ pexpect==4.9.0
65
+ pillow==10.2.0
66
+ platformdirs==4.2.0
67
+ prompt-toolkit==3.0.43
68
+ psutil==5.9.8
69
+ ptyprocess==0.7.0
70
+ pure-eval==0.2.2
71
+ py-cpuinfo==9.0.0
72
+ pydantic==2.6.1
73
+ pydantic_core==2.16.2
74
+ pydub==0.25.1
75
+ Pygments==2.17.2
76
+ PyJWT==2.8.0
77
+ pyparsing==3.1.1
78
+ PySocks==1.7.1
79
+ python-dateutil==2.8.2
80
+ python-dotenv==1.0.1
81
+ python-magic==0.4.27
82
+ python-multipart==0.0.9
83
+ pytz==2024.1
84
+ PyYAML==6.0.1
85
+ pyzmq==25.1.2
86
+ referencing==0.33.0
87
+ requests==2.31.0
88
+ requests-toolbelt==1.0.0
89
+ rich==13.7.0
90
+ roboflow==1.1.19
91
+ rpds-py==0.18.0
92
+ ruff==0.2.2
93
+ scipy==1.12.0
94
+ seaborn==0.13.2
95
+ semantic-version==2.10.0
96
+ shellingham==1.5.4
97
+ six==1.16.0
98
+ sniffio==1.3.0
99
+ soupsieve==2.5
100
+ stack-data==0.6.3
101
+ starlette==0.36.3
102
+ supervision==0.18.0
103
+ sympy==1.12
104
+ thop==0.1.1.post2209072238
105
+ tomlkit==0.12.0
106
+ toolz==0.12.1
107
+ torch==2.2.0
108
+ torchvision==0.17.0
109
+ tornado==6.4
110
+ tqdm==4.66.2
111
+ traitlets==5.14.1
112
+ typer==0.9.0
113
+ typing_extensions==4.9.0
114
+ tzdata==2024.1
115
+ ultralytics==8.1.11
116
+ urllib3==2.2.0
117
+ uvicorn==0.27.1
118
+ wcwidth==0.2.13
119
+ websockets==11.0.3