Only resize

2024-06-16 16:17:38 +05:00
parent f3d84c4545
commit 65a8fdcbf8
1 changed files with 1 additions and 1 deletions
@@ -56,7 +56,7 @@ class YOLOClassify:

    def classify(self, yolo_classifier_model, image_to_classify):
        transform = transforms.Compose(
-            [transforms.ToTensor(), transforms.Resize(size=(640, 640))]
+            [transforms.Resize(size=(640, 640))]
        )

        img_tensor = transform(image_to_classify).unsqueeze(0)