Nonabzbssbbsbs commited on
Commit
69f9057
·
verified ·
1 Parent(s): fbb9909

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
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
- # detected object-терді тексеру
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: # басқа detected object-терді қоқыс деп қарау
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