NextGenC commited on
Commit
5a25efb
·
verified ·
1 Parent(s): eca2140

Upload 2 files

Browse files
Files changed (2) hide show
  1. next_vision.py +67 -0
  2. yolov8x.pt +3 -0
next_vision.py ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import cv2
2
+ import torch
3
+ from ultralytics import YOLO
4
+
5
+ # CUDA kullanılabilirliğini kontrol et
6
+ device = "cuda" if torch.cuda.is_available() else "cpu"
7
+ print(f"Çalıştırılan cihaz: {device}")
8
+
9
+ # YOLOv8x modelini yükle
10
+ model = YOLO('yolov8x.pt')
11
+
12
+ # Bilgisayar kamerasını başlat
13
+ cap = cv2.VideoCapture(0)
14
+
15
+ if not cap.isOpened():
16
+ print("Kamera açılamadı. Lütfen kameranın bağlı olduğundan emin olun.")
17
+ exit()
18
+
19
+ print("Anlık nesne tespiti başlatıldı. Kamerayı kapatmak için 'q' tuşuna basın.")
20
+
21
+ try:
22
+ while True:
23
+ ret, frame = cap.read()
24
+ if not ret:
25
+ print("Kamera akışında hata oluştu.")
26
+ break
27
+
28
+ # Model ile nesne tespiti yap
29
+ results = model.predict(frame, device=device, conf=0.5, show=False)
30
+
31
+ # Tespit edilen nesneleri terminale yazdır ve görüntü üzerinde göster
32
+ detected_objects = [] # Tespit edilen nesneleri saklamak için liste
33
+ for result in results:
34
+ for box in result.boxes:
35
+ label = model.names[int(box.cls)] # Etiket adı
36
+ confidence = float(box.conf) # Güven skoru float'a dönüştürülüyor
37
+ detected_objects.append((label, confidence))
38
+
39
+ # Kutu bilgilerini al ve görüntü üzerine çiz
40
+ x1, y1, x2, y2 = map(int, box.xyxy[0]) # Koordinatları int'e dönüştür
41
+ cv2.rectangle(frame, (x1, y1), (x2, y2), (0, 255, 0), 2)
42
+ cv2.putText(frame, f"{label} {confidence:.2f}", (x1, y1 - 10),
43
+ cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2)
44
+
45
+ if detected_objects:
46
+ print("Tespit edilen nesneler:")
47
+ for obj in detected_objects:
48
+ print(f"- Nesne: {obj[0]}, Skor: {obj[1]:.2f}")
49
+ else:
50
+ print("Nesne tespit edilmedi.")
51
+
52
+ # Görüntüyü göster
53
+ cv2.imshow("Next Vision", frame)
54
+
55
+ # 'q' tuşuna basıldığında döngüden çık
56
+ if cv2.waitKey(1) & 0xFF == ord('q'):
57
+ print("Kapatılıyor...")
58
+ break
59
+
60
+ except KeyboardInterrupt:
61
+ print("Kullanıcı tarafından durduruldu.")
62
+
63
+ finally:
64
+ # Kaynakları serbest bırak
65
+ cap.release()
66
+ cv2.destroyAllWindows()
67
+ print("Program sonlandırıldı.")
yolov8x.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3df4ada6b4dad6d657868f2fdf7faecfb34dcfccf3a25c4b82079064718524c8
3
+ size 136890692