test

uaarg · Feb 6, 2024 · 2633144 · 2633144
1 parent c0bcdf2
commit 2633144
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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