From 1cb337c7b14084f4fe2d3c64dc3b0dd04e454957 Mon Sep 17 00:00:00 2001
From: Alexander Popov <iiiypuk@fastmail.fm>
Date: Sat, 24 Feb 2024 19:30:01 +0300
Subject: [PATCH] =?UTF-8?q?=D1=85=D1=83=D0=B9=20=D0=B7=D0=BD=D0=B0=D0=B5?=
 =?UTF-8?q?=D1=82=20=D1=87=D1=82=D0=BE=20=D1=82=D0=B0=D0=BC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 projects/CV/ex.py | 24 ++++++++++++++----------
 1 file changed, 14 insertions(+), 10 deletions(-)

diff --git a/projects/CV/ex.py b/projects/CV/ex.py
index 9a656bf..3576f9b 100644
--- a/projects/CV/ex.py
+++ b/projects/CV/ex.py
@@ -1,23 +1,27 @@
 from ultralytics import YOLO
 
-model = YOLO('yolov8m.pt')
+model = YOLO('yolov8s.pt')
 model.save_dir = './predict/'
 
 result = model.predict(
     './Assets/cat_dog.jpg',
     save=True,
-    device='cpu',
+    project='test',
+    name='ololo',
+    device=0,
     show=False,
     classes=[16, 15],
 )[0]
-result = model(
-    './Assets/cat_dog.jpg',
-    save=True,
-    project='test',
-    name='ololo',
-    device='cpu',
-    show=False,
-)[0]
+# result = model(
+#     './Assets/cat_dog.jpg',
+#     save=True,
+#     project='test',
+#     name='ololo',
+#     device='gpu',
+#     show=False,
+# )[0]
+
+print(model.names)
 
 # print(len(result.boxes))