Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,23 +8,21 @@ model = YOLO('yolov8n.pt')
|
|
| 8 |
# Қоқыс жәшігін тексеру және дабыл шығару функциясы
|
| 9 |
def check_trash(img):
|
| 10 |
results = model(img)
|
| 11 |
-
boxes = results[0].boxes
|
| 12 |
detected_bins = []
|
| 13 |
detected_trash = []
|
| 14 |
|
| 15 |
-
|
| 16 |
-
for box in boxes:
|
| 17 |
label = results[0].names[int(box.cls.item())]
|
| 18 |
conf = box.conf.item()
|
| 19 |
if conf < 0.5:
|
| 20 |
continue
|
| 21 |
-
if 'bin' in label or 'container' in label: #
|
| 22 |
detected_bins.append(label)
|
| 23 |
-
else: #
|
| 24 |
detected_trash.append(label)
|
| 25 |
|
| 26 |
-
# Нәтижені шығару
|
| 27 |
output = []
|
|
|
|
| 28 |
if detected_bins:
|
| 29 |
if detected_trash:
|
| 30 |
output.append("Жәшік: Толы ✅")
|
|
@@ -33,6 +31,7 @@ def check_trash(img):
|
|
| 33 |
else:
|
| 34 |
output.append("Жәшік табылмады ❌")
|
| 35 |
|
|
|
|
| 36 |
if detected_trash:
|
| 37 |
output.append(f"Сыртқа шыққан қоқыс табылды 🚨: {', '.join(detected_trash)}")
|
| 38 |
|
|
|
|
| 8 |
# Қоқыс жәшігін тексеру және дабыл шығару функциясы
|
| 9 |
def check_trash(img):
|
| 10 |
results = model(img)
|
|
|
|
| 11 |
detected_bins = []
|
| 12 |
detected_trash = []
|
| 13 |
|
| 14 |
+
for box in results[0].boxes:
|
|
|
|
| 15 |
label = results[0].names[int(box.cls.item())]
|
| 16 |
conf = box.conf.item()
|
| 17 |
if conf < 0.5:
|
| 18 |
continue
|
| 19 |
+
if 'bin' in label or 'container' in label: # Жәшік
|
| 20 |
detected_bins.append(label)
|
| 21 |
+
else: # Қоқыс
|
| 22 |
detected_trash.append(label)
|
| 23 |
|
|
|
|
| 24 |
output = []
|
| 25 |
+
# Толы / бос жәшік
|
| 26 |
if detected_bins:
|
| 27 |
if detected_trash:
|
| 28 |
output.append("Жәшік: Толы ✅")
|
|
|
|
| 31 |
else:
|
| 32 |
output.append("Жәшік табылмады ❌")
|
| 33 |
|
| 34 |
+
# Сыртқа шыққан қоқыс
|
| 35 |
if detected_trash:
|
| 36 |
output.append(f"Сыртқа шыққан қоқыс табылды 🚨: {', '.join(detected_trash)}")
|
| 37 |
|