From 5cc2661ca3ee34e6dac74ad462f5ce3aa64d94d3 Mon Sep 17 00:00:00 2001
From: "DESKTOP-UL3K6JV\\Daniq" <31.10.93@bk.ru>
Date: Sun, 16 Jun 2024 17:19:54 +0500
Subject: [PATCH] Size to 224

---
 classifiers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/classifiers.py b/classifiers.py
index c0a885b..81aa98b 100644
--- a/classifiers.py
+++ b/classifiers.py
@@ -58,7 +58,7 @@ class YOLOClassify:
         image_to_classify = image_to_classify.permute(0, 3, 1, 2)
         image_to_classify = image_to_classify.squeeze(0)
         transform = transforms.Compose([
-            transforms.Resize(size=(640, 640))
+            transforms.Resize(size=(224, 224))
         ])
 
         img = transform(image_to_classify).unsqueeze(0)