server.py added for getting feed from online link

Author: TeleVision05

.gitignore

@@ -0,0 +1 @@
+software/cricket.pt

README.md

@@ -1 +1,11 @@
 # Security Camera Team 3
+
+
+MQTT Message Format:
+
+objects = [{'class': classes[i], 'confidence': confidences[i]} for i in range(len(classes))]
+ 
+message = {
+ "objects": objects,
+ "timestamp": datetime.now().isoformat()
+}

software/50epoch.pt

@@ -0,0 +1,119 @@
+"""
+MQTT Client Module for Security Camera
+Handles MQTT connection and message publishing for object detection alerts
+"""
+import paho.mqtt.client as mqtt
+import json
+from datetime import datetime
+import threading
+import time
+
+# MQTT Configuration
+MQTT_BROKER = "mqtt.a5-p.cn"
+MQTT_PORT = 1883
+MQTT_USERNAME = "forge"
+MQTT_PASSWORD = "zDHRPjrH1G5iETx"
+MQTT_TOPIC = "security_camera/detections"
+
+class MQTTClient:
+ """MQTT Client for sending detection messages"""
+ 
+ def __init__(self):
+ self.client = None
+ self.connected = False
+ self.connection_lock = threading.Lock()
+ 
+ def setup_connection(self):
+ """Setup MQTT client and establish connection"""
+ try:
+ self.client = mqtt.Client()
+ self.client.username_pw_set(MQTT_USERNAME, MQTT_PASSWORD)
+ 
+ # Set up callbacks
+ self.client.on_connect = self._on_connect
+ self.client.on_disconnect = self._on_disconnect
+ self.client.on_publish = self._on_publish
+ 
+ # Connect to broker
+ self.client.connect(MQTT_BROKER, MQTT_PORT, 60)
+ self.client.loop_start()
+ 
+ # Wait a moment for connection to establish
+ time.sleep(1)
+ return self.connected
+ 
+ except Exception as e:
+ print(f"❌ Error setting up MQTT connection: {e}")
+ return False
+ 
+ def _on_connect(self, client, userdata, flags, rc):
+ """Callback for when client connects to broker"""
+ if rc == 0:
+ self.connected = True
+ print("✅ Connected to MQTT broker")
+ else:
+ self.connected = False
+ print(f"❌ Failed to connect to MQTT broker. Return code: {rc}")
+ 
+ def _on_disconnect(self, client, userdata, rc):
+ """Callback for when client disconnects from broker"""
+ self.connected = False
+ print("Disconnected from MQTT broker")
+ 
+ def _on_publish(self, client, userdata, mid):
+ """Callback for when message is published"""
+ print(f"📤 Message published successfully (mid: {mid})")
+ 
+ def send_detection(self, classes, confidences, source="rtmp_stream"):
+ """Send object detection message to MQTT broker"""
+ if not self.connected or self.client is None:
+ print("❌ MQTT client not connected. Cannot send message.")
+ return False
+ 
+ objects = [{'class': classes[i], 'confidence': confidences[i]} for i in range(len(classes))]
+ # Create detection message
+ message = {
+ "objects": objects,
+ "timestamp": datetime.now().isoformat()
+ }
+ 
+ try:
+ # Publish message
+ result = self.client.publish(MQTT_TOPIC, json.dumps(message))
+ 
+ if result.rc == mqtt.MQTT_ERR_SUCCESS:
+ print(f"📤 Sent detection: {objects}")
+ return True
+ else:
+ print(f"❌ Failed to publish message. Return code: {result.rc}")
+ return False
+ 
+ except Exception as e:
+ print(f"❌ Error sending MQTT message: {e}")
+ return False
+ 
+ def disconnect(self):
+ """Disconnect from MQTT broker"""
+ if self.client:
+ self.client.loop_stop()
+ self.client.disconnect()
+ print("Disconnected from MQTT broker")
+
+# Global MQTT client instance
+mqtt_client = MQTTClient()
+
+def initialize_mqtt():
+ """Initialize MQTT connection"""
+ return mqtt_client.setup_connection()
+
+def send_detection_message(class_name, confidence, source="rtmp_stream"):
+ """Send detection message using global MQTT client"""
+ return mqtt_client.send_detection(class_name, confidence, source)
+
+def disconnect_mqtt():
+ """Disconnect from MQTT broker"""
+ mqtt_client.disconnect()
+
+def is_mqtt_connected():
+ """Check if MQTT client is connected"""
+ return mqtt_client.connected

software/__pycache__/mqtt_client.cpython-313.pyc

@@ -0,0 +1,83 @@
+import cv2
+import argparse
+from ultralytics import YOLO
+from mqtt_client import initialize_mqtt, send_detection_message, is_mqtt_connected
+
+conversion = ["car", "cat", "dog", "person"]
+skip_frames = 3
+def score_to_bgr(score: float) -> tuple[int, int, int]:
+ """Map score in [0,1] to a BGR color from red (low) to green (high)."""
+ s = max(0.0, min(1.0, float(score)))
+ g = int(255 * s)
+ r = int(255 * (1.0 - s))
+ b = 0
+ return (b, g, r)
+
+
+def run_webcam(model, img_size):
+ frames = skip_frames
+ 
+ # Initialize MQTT connection
+ print("🔌 Initializing MQTT connection...")
+ if initialize_mqtt():
+ print("✅ MQTT connection established")
+ else:
+ print("⚠️ MQTT connection failed, continuing without MQTT")
+
+ cap = cv2.VideoCapture("rtmps://live.cloudflare.com:443/live/25f7b82374cb80506bafaa1c45e4e70ek7690d120ab117d3b6e1cbcfc3ab3e758")
+ if not cap.isOpened():
+ print("Could not open webcam.")
+ return
+ print("Press 'q' to quit.")
+ while True:
+ ret, frame = cap.read()
+ if not ret:
+ break
+ if frames > 0:
+ frames -= 1
+ continue
+ else:
+ frames = skip_frames
+ results = model(frame, imgsz=img_size)
+ for r in results:
+ boxes = r.boxes.xyxy.cpu().numpy()
+ scores = r.boxes.conf.cpu().numpy()
+ labels = r.boxes.cls.cpu().numpy().astype(int)
+ for (x1, y1, x2, y2), score, label in zip(boxes, scores, labels):
+ if score < 0.6:
+ continue
+ color = score_to_bgr(score)
+ text = f"{conversion[int(model.names[label])]}: {score:.2f}" if hasattr(model, 'names') else f"{label}: {score:.2f}"
+
+ # Get class name from model
+ class_name = model.names[label] if hasattr(model, 'names') and label in model.names else f"class_{label}"
+ text = f"{class_name}: {score:.2f}"
+ 
+ # Send MQTT message for detected object
+ if is_mqtt_connected():
+ send_detection_message(class_name, score, "rtmp_stream")
+ 
+ # draw box
+ cv2.rectangle(frame, (int(x1), int(y1)), (int(x2), int(y2)), color, 2)
+ # text background for readability
+ (tw, th), _ = cv2.getTextSize(text, cv2.FONT_HERSHEY_SIMPLEX, 0.5, 1)
+ tx1 = int(x1)
+ ty1 = int(y1) - th - 6
+ ty1 = max(0, ty1)
+ tx2 = tx1 + tw + 6
+ ty2 = ty1 + th + 6
+ cv2.rectangle(frame, (tx1, ty1), (tx2, ty2), color, -1)
+ cv2.putText(frame, text, (tx1 + 3, ty2 - 4), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 0), 1)
+ cv2.imshow('webcam', frame)
+ if cv2.waitKey(1) & 0xFF == ord('q'):
+ break
+ cap.release()
+ cv2.destroyAllWindows()
+
+if __name__ == '__main__':
+ parser = argparse.ArgumentParser()
+ parser.add_argument('--model', default='50epoch.pt')
+ parser.add_argument('--img-size', type=int, nargs=2, default=(320, 320))
+ args = parser.parse_args()
+ model = YOLO(args.model)
+ run_webcam(model, tuple(args.img_size))