mirror of
https://gitlab.fdmci.hva.nl/technische-informatica-sm3/ti-projectten/rooziinuubii79.git
synced 2025-08-05 12:54:57 +00:00
added yolo image object detection to /image
This commit is contained in:
41
src/Python/YOLO/app.py
Normal file
41
src/Python/YOLO/app.py
Normal file
@@ -0,0 +1,41 @@
|
||||
from ultralytics import YOLO
|
||||
import cv2
|
||||
import numpy as np
|
||||
import requests
|
||||
import time
|
||||
|
||||
model = YOLO("yolo11n.pt")
|
||||
|
||||
#try to fetch the image from the given url
|
||||
def fetch_image(url):
|
||||
try:
|
||||
response = requests.get(url)
|
||||
response.raise_for_status()
|
||||
image_array = np.frombuffer(response.content, np.uint8)
|
||||
image = cv2.imdecode(image_array, cv2.IMREAD_COLOR)
|
||||
return image
|
||||
except requests.RequestException as e:
|
||||
print(f"Error: Could not fetch image - {e}")
|
||||
return None
|
||||
|
||||
# URL of the photostream
|
||||
url = "http://145.92.224.21/image"
|
||||
|
||||
while True:
|
||||
frame = fetch_image(url)
|
||||
if frame is None:
|
||||
print("Error: Could not fetch image, retrying...")
|
||||
time.sleep(1) # Wait for 1 second before retrying
|
||||
continue
|
||||
|
||||
# Predict on the frame
|
||||
results = model(frame)
|
||||
|
||||
# Display the results
|
||||
results[0].show()
|
||||
|
||||
# Exit if 'q' is pressed
|
||||
if cv2.waitKey(1) & 0xFF == ord('q'):
|
||||
break
|
||||
|
||||
cv2.destroyAllWindows()
|
Reference in New Issue
Block a user