From 2633144bf4c978054dcd37214d177dbd694b79a8 Mon Sep 17 00:00:00 2001 From: fadilm777 Date: Mon, 5 Feb 2024 17:26:22 -0700 Subject: [PATCH] test --- benchmarks/yolov8.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/benchmarks/yolov8.py b/benchmarks/yolov8.py index 7319f7b..e56f42d 100644 --- a/benchmarks/yolov8.py +++ b/benchmarks/yolov8.py @@ -42,7 +42,7 @@ def predict(self, image: Image.Image) -> Optional[BoundingBox]: Returns a list with dictionary items for each prediction """ im0 = np.array(image) - print(im0) + print(im0[0]) @@ -56,7 +56,7 @@ def predict(self, image: Image.Image) -> Optional[BoundingBox]: device=self.device, max_det=self.max_det, ) - print(results) + print(results[0]) if __name__ == "__main__": import glob