From c6bafafa129b5e177cd5fa59334ed53a32e36ce5 Mon Sep 17 00:00:00 2001
From: "DESKTOP-UL3K6JV\\Daniq" <31.10.93@bk.ru>
Date: Sun, 16 Jun 2024 17:57:03 +0500
Subject: [PATCH] half

---
 classifiers.py | 7 +------
 1 file changed, 1 insertion(+), 6 deletions(-)

diff --git a/classifiers.py b/classifiers.py
index f190670..4862afa 100644
--- a/classifiers.py
+++ b/classifiers.py
@@ -56,12 +56,7 @@ class YOLOClassify:
 
     def classify(self, yolo_classifier_model, image_to_classify):
         image_to_classify = image_to_classify.permute(0, 3, 1, 2)
-
-        if yolo_classifier_model.model.fp16:
-            image_to_classify = image_to_classify.half()
-        else:
-            image_to_classify = image_to_classify.float()
-
+        image_to_classify = image_to_classify.half()
         image_to_classify = image_to_classify.squeeze(0)
         transform = transforms.Compose([
             transforms.Resize(size=(224, 224))