fix: adjust resize mode

Author: Joker1212

table_cls/main.py

@@ -5,17 +5,20 @@
 import numpy as np
 from PIL import Image
 
-from .utils import InputType, LoadImage, OrtInferSession, resize_and_center_crop
+from .utils import InputType, LoadImage, OrtInferSession
 
 cur_dir = Path(__file__).resolve().parent
 q_cls_model_path = cur_dir / "models" / "table_cls.onnx"
 yolo_cls_model_path = cur_dir / "models" / "yolo_cls.onnx"
+yolo_cls_x_model_path = cur_dir / "models" / "yolo_cls_x.onnx"
 
 
 class TableCls:
     def __init__(self, model_type="yolo", model_path=yolo_cls_model_path):
         if model_type == "yolo":
             self.table_engine = YoloCls(model_path)
+        elif model_type == "yolox":
+            self.table_engine = YoloCls(yolo_cls_x_model_path)
         else:
             model_path = q_cls_model_path
             self.table_engine = QanythingCls(model_path)
@@ -64,15 +67,11 @@ class YoloCls:
     def __init__(self, model_path):
         self.table_cls = OrtInferSession(model_path)
         self.cls = {0: "wireless", 1: "wired"}
-        self.mean = np.array([0, 0, 0], dtype=np.float32)
-        self.std = np.array([1, 1, 1], dtype=np.float32)
 
     def preprocess(self, img):
         img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
-        img = resize_and_center_crop(img, 640)
+        img = cv2.resize(img, (640, 640))
         img = np.array(img, dtype=np.float32) / 255
-        img -= self.mean
-        img /= self.std
         img = img.transpose(2, 0, 1)  # HWC to CHW
         img = np.expand_dims(img, axis=0)  # Add batch dimension, only one image
         return img