From 75305865df5d2f3c9d8061d7f405f6a41f5e9a9e Mon Sep 17 00:00:00 2001 From: Ishaan Datta Date: Fri, 27 Sep 2024 09:34:57 -0700 Subject: [PATCH] engine test v1 --- conversion_tools/ultralytics_test_2.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) create mode 100644 conversion_tools/ultralytics_test_2.py diff --git a/conversion_tools/ultralytics_test_2.py b/conversion_tools/ultralytics_test_2.py new file mode 100644 index 0000000..364ecd9 --- /dev/null +++ b/conversion_tools/ultralytics_test_2.py @@ -0,0 +1,22 @@ +import cv2 +from ultralytics import YOLO +import time + +model = YOLO("/home/user/ROS/models/maize/Maize.engine") +image = cv2.imread("/home/user/ROS/assets/maize/IMG_1822_14.JPG") + +# stream = True? +for _ in range(100): + tic = time.perf_counter_ns() + result = model.predict( + image, # batch=8 of the same image + verbose=False, + device="cuda", + ) + print(f"Elapsed time: {(time.perf_counter_ns() - tic) / 1e6:.2f} ms") + annotated_frame = result[0].plot() + cv2.imshow("YOLOv8 Inference", annotated_frame) + if cv2.waitKey(1) & 0xFF == ord("q"): + break + +cv2.destroyAllWindows() \ No newline at end of file