Size to 224

2024-06-16 17:19:54 +05:00
parent d449ff8f98
commit 5cc2661ca3
1 changed files with 1 additions and 1 deletions
@@ -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)