Spaces:
Sleeping
Sleeping
aaaaaa
Browse files
app.py
CHANGED
|
@@ -125,4 +125,162 @@ def process_stream_frame(frame):
|
|
| 125 |
except Empty:
|
| 126 |
return frame
|
| 127 |
|
| 128 |
-
#
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 125 |
except Empty:
|
| 126 |
return frame
|
| 127 |
|
| 128 |
+
# ๋ฉ์ธ ์ฑ
|
| 129 |
+
with gr.Blocks(title="์ด๊ณ ์ YOLO ์คํธ๋ฆฌ๋ฐ") as app:
|
| 130 |
+
gr.Markdown("""
|
| 131 |
+
# โก ์ด๊ณ ์ YOLO ์คํธ๋ฆฌ๋ฐ ํ์ง
|
| 132 |
+
|
| 133 |
+
**์๋ ์ต์ ํ ๊ธฐ๋ฒ๋ค:**
|
| 134 |
+
- YOLOv5n ์ฌ์ฉ (๊ฐ์ฅ ๋น ๋ฅธ ๋ชจ๋ธ)
|
| 135 |
+
- ์์ ์
๋ ฅ ํฌ๊ธฐ (320x320, 160x160)
|
| 136 |
+
- FP16 ๋ฐ์ ๋ฐ๋ ์ฐ์ฐ
|
| 137 |
+
- ํ ํฌ๊ธฐ ์ต์ํ
|
| 138 |
+
- ํ๋ ์ ๋๋กญ ํ์ฉ
|
| 139 |
+
""")
|
| 140 |
+
|
| 141 |
+
with gr.Tabs():
|
| 142 |
+
with gr.TabItem("๐ ๊ณ ์ ์คํธ๋ฆฌ๋ฐ"):
|
| 143 |
+
with gr.Row():
|
| 144 |
+
with gr.Column():
|
| 145 |
+
webcam_input = gr.Image(
|
| 146 |
+
label="๐น ์น์บ ์
๋ ฅ",
|
| 147 |
+
sources=["webcam"],
|
| 148 |
+
streaming=True,
|
| 149 |
+
width=480,
|
| 150 |
+
height=360
|
| 151 |
+
)
|
| 152 |
+
|
| 153 |
+
with gr.Row():
|
| 154 |
+
start_btn = gr.Button("๐ ์์", variant="primary")
|
| 155 |
+
stop_btn = gr.Button("โน๏ธ ์ค์ง", variant="secondary")
|
| 156 |
+
|
| 157 |
+
with gr.Accordion("โ๏ธ ์ค์ ", open=True):
|
| 158 |
+
model_select = gr.Dropdown(
|
| 159 |
+
choices=list(FAST_MODELS.keys()) + ["์
๋ก๋๋ ํ์ผ"],
|
| 160 |
+
value="yolov5n",
|
| 161 |
+
label="๋ชจ๋ธ ์ ํ (์์์๋ถํฐ ๋น ๋ฆ)"
|
| 162 |
+
)
|
| 163 |
+
|
| 164 |
+
weight_upload = gr.File(
|
| 165 |
+
label="๊ฐ์ค์น ์
๋ก๋",
|
| 166 |
+
file_types=[".pt"],
|
| 167 |
+
visible=False
|
| 168 |
+
)
|
| 169 |
+
|
| 170 |
+
conf_threshold = gr.Slider(
|
| 171 |
+
0.1, 1.0, 0.3, 0.1,
|
| 172 |
+
label="์ ๋ขฐ๋ ์๊ณ๊ฐ"
|
| 173 |
+
)
|
| 174 |
+
|
| 175 |
+
iou_threshold = gr.Slider(
|
| 176 |
+
0.1, 1.0, 0.5, 0.1,
|
| 177 |
+
label="IoU ์๊ณ๊ฐ"
|
| 178 |
+
)
|
| 179 |
+
|
| 180 |
+
with gr.Column():
|
| 181 |
+
stream_output = gr.Image(
|
| 182 |
+
label="๐ฏ ํ์ง ๊ฒฐ๊ณผ",
|
| 183 |
+
width=480,
|
| 184 |
+
height=360
|
| 185 |
+
)
|
| 186 |
+
|
| 187 |
+
status_display = gr.Textbox(
|
| 188 |
+
label="๐ ์ํ",
|
| 189 |
+
value="์ค๋น ์๋ฃ",
|
| 190 |
+
interactive=False
|
| 191 |
+
)
|
| 192 |
+
|
| 193 |
+
with gr.TabItem("โก ๊ทนํ ์ต์ ํ"):
|
| 194 |
+
gr.Markdown("""
|
| 195 |
+
### ๊ทนํ ์ต์ ํ ๋ชจ๋
|
| 196 |
+
- 160x160 ์ด์ํ ์ฒ๋ฆฌ
|
| 197 |
+
- ์ต์ํ์ ์๊ฐํ
|
| 198 |
+
- ํ๋ ์ ๋๋กญ ํ์ฉ
|
| 199 |
+
- ๋ชฉํ: 60+ FPS
|
| 200 |
+
""")
|
| 201 |
+
|
| 202 |
+
with gr.Row():
|
| 203 |
+
ultra_input = gr.Image(
|
| 204 |
+
label="๐น ๊ทนํ ์ต์ ํ ์
๋ ฅ",
|
| 205 |
+
sources=["webcam"],
|
| 206 |
+
streaming=True,
|
| 207 |
+
width=320,
|
| 208 |
+
height=240
|
| 209 |
+
)
|
| 210 |
+
|
| 211 |
+
ultra_output = gr.Image(
|
| 212 |
+
label="โก ๊ทนํ ์ต์ ํ ์ถ๋ ฅ",
|
| 213 |
+
width=320,
|
| 214 |
+
height=240
|
| 215 |
+
)
|
| 216 |
+
|
| 217 |
+
# ์ด๋ฒคํธ ๋ฐ์ธ๋ฉ
|
| 218 |
+
def toggle_upload(choice):
|
| 219 |
+
return gr.update(visible=(choice == "์
๋ก๋๋ ํ์ผ"))
|
| 220 |
+
|
| 221 |
+
model_select.change(
|
| 222 |
+
fn=toggle_upload,
|
| 223 |
+
inputs=[model_select],
|
| 224 |
+
outputs=[weight_upload]
|
| 225 |
+
)
|
| 226 |
+
|
| 227 |
+
start_btn.click(
|
| 228 |
+
fn=start_streaming,
|
| 229 |
+
inputs=[model_select, weight_upload, conf_threshold, iou_threshold],
|
| 230 |
+
outputs=[status_display]
|
| 231 |
+
)
|
| 232 |
+
|
| 233 |
+
stop_btn.click(
|
| 234 |
+
fn=stop_streaming,
|
| 235 |
+
outputs=[status_display]
|
| 236 |
+
)
|
| 237 |
+
|
| 238 |
+
# ์คํธ๋ฆผ ์ฒ๋ฆฌ
|
| 239 |
+
webcam_input.stream(
|
| 240 |
+
fn=process_stream_frame,
|
| 241 |
+
inputs=[webcam_input],
|
| 242 |
+
outputs=[stream_output],
|
| 243 |
+
stream_every=0.02 # 50 FPS ๋ชฉํ
|
| 244 |
+
)
|
| 245 |
+
|
| 246 |
+
# ๊ทนํ ์ต์ ํ ์คํธ๋ฆผ
|
| 247 |
+
ultra_input.stream(
|
| 248 |
+
fn=ultra_fast_detect,
|
| 249 |
+
inputs=[ultra_input],
|
| 250 |
+
outputs=[ultra_output],
|
| 251 |
+
stream_every=0.016 # 60 FPS ๋ชฉํ
|
| 252 |
+
)
|
| 253 |
+
|
| 254 |
+
gr.Markdown("""
|
| 255 |
+
---
|
| 256 |
+
## ๐โโ๏ธ ์๋ ๊ฐ์ ํ
|
| 257 |
+
|
| 258 |
+
### 1. ๋ชจ๋ธ ์ ํ (๋น ๋ฅธ ์์)
|
| 259 |
+
- **YOLOv5n**: ๊ฐ์ฅ ๋น ๋ฆ, ์ ํ๋ ์ ๋น
|
| 260 |
+
- **YOLOv5s**: ๋น ๋ฆ, ์ ํ๋ ์ข์
|
| 261 |
+
- **YOLOv8n**: ๋ณดํต ์๋, ๋์ ์ ํ๋
|
| 262 |
+
|
| 263 |
+
### 2. ํ๋์จ์ด ์ต์ ํ
|
| 264 |
+
- GPU ์ฌ์ฉ (CUDA)
|
| 265 |
+
- ์ถฉ๋ถํ RAM
|
| 266 |
+
- ์น์บ ํด์๋ ๋ฎ์ถ๊ธฐ
|
| 267 |
+
|
| 268 |
+
### 3. ์ค์ ์ต์ ํ
|
| 269 |
+
- ๋์ confidence ์๊ณ๊ฐ (0.5+)
|
| 270 |
+
- ์์ ์
๋ ฅ ํฌ๊ธฐ
|
| 271 |
+
- ๋ถํ์ํ ํ์ฒ๋ฆฌ ์ ๊ฑฐ
|
| 272 |
+
|
| 273 |
+
### 4. ์์คํ
์ต์ ํ
|
| 274 |
+
- ๋ธ๋ผ์ฐ์ ๋ค๋ฅธ ํญ ๋ซ๊ธฐ
|
| 275 |
+
- ๋ฐฑ๊ทธ๋ผ์ด๋ ํ๋ก๊ทธ๋จ ์ข
๋ฃ
|
| 276 |
+
- ์น์บ ๋๋ผ๏ฟฝ๏ฟฝ๏ฟฝ๋ฒ ์
๋ฐ์ดํธ
|
| 277 |
+
""")
|
| 278 |
+
|
| 279 |
+
if __name__ == "__main__":
|
| 280 |
+
app.launch(
|
| 281 |
+
share=True,
|
| 282 |
+
server_name="0.0.0.0",
|
| 283 |
+
server_port=7860,
|
| 284 |
+
show_error=True,
|
| 285 |
+
debug=True
|
| 286 |
+
)
|