start CV

2023-09-29 21:51:53 +03:00
parent f8b9ae2443
commit cde52c4596
13 changed files with 137 additions and 0 deletions
@@ -0,0 +1,28 @@
+from ultralytics import YOLO
+
+model = YOLO('yolov8m.pt')
+model.save_dir = './predict/'
+
+result = model.predict(
+    './Assets/cat_dog.jpg',
+    save=True,
+    device='cpu',
+    show=False,
+    classes=[16, 15],
+)[0]
+result = model(
+    './Assets/cat_dog.jpg',
+    save=True,
+    project='test',
+    name='ololo',
+    device='cpu',
+    show=False,
+)[0]
+
+# print(len(result.boxes))
+
+# for _ in result.boxes:
+#     box = result.boxes[_]
+#     print("Object type:", box.cls)
+#     print("Coordinates:", box.xyxy)
+# print("Probability:", box.conf)