Added division by 255

2024-06-16 17:31:21 +05:00
parent 5cc2661ca3
commit 9fb37f950a
1 changed files with 1 additions and 0 deletions
@@ -56,6 +56,7 @@ class YOLOClassify:

    def classify(self, yolo_classifier_model, image_to_classify):
        image_to_classify = image_to_classify.permute(0, 3, 1, 2)
+        image_to_classify = image_to_classify / 255
        image_to_classify = image_to_classify.squeeze(0)
        transform = transforms.Compose([
            transforms.Resize(size=(224, 224))