consolidate all repos to one for archive

projektna_naloga/capture_raw_camera_data/.gitignore

@@ -0,0 +1 @@
+*.pt

projektna_naloga/capture_raw_camera_data/main.py

@@ -0,0 +1,102 @@
+import json
+import requests
+import cv2
+import numpy as np
+from ultralytics import YOLO
+from datetime import datetime, timedelta
+import time
+import pytz
+import threading
+
+# 2 car
+# 3 motorbike
+# 5 bus
+# 7 truck
+
+lock = threading.Lock()
+
+def worker():
+
+    global run
+    global vd
+    send_url = 'http://localhost:3000/api/rawCameraData'
+    count_url = 'https://school.petrovv.com/api/counter'
+    i = 0
+    errors = 0
+    while True:
+
+        if errors > 5:
+            break
+
+        with lock:
+            if run != True:
+                break
+            
+        try:
+            #start = time.monotonic_ns() / 1000000
+            response = requests.get(count_url)
+            cameraData = response.json()
+            cameraData = cameraData['camera']
+
+
+            response = requests.get(cameraData['img'])
+            response.raise_for_status()
+            arr = np.asarray(bytearray(response.content), dtype=np.uint8)
+            img = cv2.imdecode(arr, cv2.IMREAD_COLOR)
+
+            #imgDownload = time.monotonic_ns() / 1000000
+
+            results = vd(img, verbose=False)
+
+            #imgSee = time.monotonic_ns() / 1000000
+
+            count = 0
+            for r in results:
+                boxes = r.boxes
+                for box in boxes:
+                    classNum = int(box.cls)
+                    if classNum == 2 or classNum == 3 or classNum == 5 or classNum == 7:
+                        count += 1
+
+            #imgCount = time.monotonic_ns() / 1000000
+
+            #print("download: {:7}, see: {:7}, count:{:7}".format(imgDownload - start, imgSee - imgDownload, imgCount - imgSee))
+            
+            
+            now_utc = datetime.now(pytz.utc) + timedelta(hours=2)
+
+            data = {'location_id': cameraData['location_id'], 'camera_id': cameraData['camera_id'], 'car_count': count, 'date':  now_utc, 'hour': now_utc.hour}
+            response = requests.post(send_url, data=data)
+#            print(data)
+            if errors > 0:
+                errors = errors - 1
+            
+        except:
+            print( 'Exception ' + str(i))
+            errors = errors + 1
+
+        i += 1
+        i = i % 1000
+
+        if i == 1:
+            print(datetime.now(pytz.utc) + timedelta(hours=2))
+     
+
+run = True
+#vd = YOLO("Yolo-Weights/yolov8x.pt")
+#vd = YOLO("Yolo-Weights/yolov8l.pt")
+#vd = YOLO("Yolo-Weights/yolov8m.pt")
+#vd = YOLO("Yolo-Weights/yolov8s.pt")
+vd = YOLO("Yolo-Weights/yolov8n.pt")
+
+t = threading.Thread(target=worker)
+
+t.start()
+
+name = input("Stop:")
+
+with lock:
+    run = False
+
+t.join()
+