From 07b0813f735fbba563913811d656176dfb9a8094 Mon Sep 17 00:00:00 2001 From: luanhailiang Date: Sat, 7 Oct 2023 18:38:22 +0800 Subject: [PATCH] result --- app.py | 3 ++- test123.py | 38 ++++++++++++++++++++++++++++++++++++++ 2 files changed, 40 insertions(+), 1 deletion(-) create mode 100644 test123.py diff --git a/app.py b/app.py index 665d848..7ea6d05 100644 --- a/app.py +++ b/app.py @@ -102,7 +102,8 @@ def worker(id,source,region=None,stream=False): print("Error opening video stream....") break continue - result = model.track(frame,device=int(id)%4,tracker='botsort.yaml') + # result = model.track(frame,device=int(id)%4,tracker='botsort.yaml') + result = model(frame) use += " track:"+str((time.time()*1000) - bgn) del(ret) del(frame) diff --git a/test123.py b/test123.py new file mode 100644 index 0000000..51fe9f9 --- /dev/null +++ b/test123.py @@ -0,0 +1,38 @@ +import cv2 +from ultralytics import YOLO + +# Load the YOLOv8 model +model = YOLO('best.pt') + +# Open the video file +video_path = "./videos/123.mp4" +cap = cv2.VideoCapture(video_path) + +cv2.namedWindow("Video", cv2.WINDOW_NORMAL) # Create a named window +cv2.resizeWindow("Video", 640, 384) # Resize this window + +# Loop through the video frames +while cap.isOpened(): + # Read a frame from the video + success, frame = cap.read() + + if success: + # Run YOLOv8 inference on the frame + results = model(frame) + + # Visualize the results on the frame + annotated_frame = results[0].plot() + + # Display the annotated frame + cv2.imshow("Video", annotated_frame) + + # Break the loop if 'q' is pressed + if cv2.waitKey(1) & 0xFF == ord("q"): + break + else: + # Break the loop if the end of the video is reached + break + +# Release the video capture object and close the display window +cap.release() +cv2.destroyAllWindows() \ No newline at end of file