Skip to content

Commit

Permalink
engine test v1
Browse files Browse the repository at this point in the history
  • Loading branch information
Ishaan-Datta committed Sep 27, 2024
1 parent 1d259de commit 7530586
Showing 1 changed file with 22 additions and 0 deletions.
22 changes: 22 additions & 0 deletions conversion_tools/ultralytics_test_2.py
Original file line number Diff line number Diff line change
@@ -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()

0 comments on commit 7530586

Please sign in to comment.