修改了很多的bug

QT_test.py

@@ -93,10 +93,10 @@ def main():
     while True:
         cmd = input().strip().upper()
         if cmd == 'IM':
-            img = cv2.imread(r"/Users/zhouchao/Library/CloudStorage/OneDrive-macrosolid/PycharmProjects/wood_color/data/data20220919/dark/rgb60.png")
+            # img = cv2.imread(r"/Users/zhouchao/Library/CloudStorage/OneDrive-macrosolid/PycharmProjects/wood_color/data/data20220919/dark/rgb60.png")
-            # img = cv2.imread(r"C:\Users\FEIJINTI\PycharmProjects\wood_color\data\data20220919\dark\rgb60.png")
+            img = cv2.imread(r"C:\Users\FEIJINTI\OneDrive\PycharmProjects\wood_color\data\316\dark\rgb60.png")
             img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
-            img = np.asarray(img, dtype=np.float32)
+            img = np.asarray(img, dtype=np.uint8)
             width = img.shape[0]
             height = img.shape[1]
             img_bytes = img.tobytes()
@@ -118,8 +118,8 @@ def main():
             # else:
             #     print('指令执行失败')
         elif cmd == 'TR':
-            model = "/Users/zhouchao/Library/CloudStorage/OneDrive-macrosolid/PycharmProjects/wood_color/data/data20220919"
+            # model = "/Users/zhouchao/Library/CloudStorage/OneDrive-macrosolid/PycharmProjects/wood_color/data/data20220919"
-            # model = "C:/Users/FEIJINTI/PycharmProjects/wood_color/data/data20220919"
+            model = r"C:\Users\FEIJINTI\OneDrive\PycharmProjects\wood_color\data\316"
             model = model.encode('ascii')
             length = len(model) + 4
             length = length.to_bytes(4, byteorder='big')
@@ -137,8 +137,8 @@ def main():
             # else:
             #     print('指令执行失败')
         elif cmd == 'MD':
-            model = "/Users/zhouchao/Library/CloudStorage/OneDrive-macrosolid/PycharmProjects/wood_color/models/model_2020-11-08_20-49.p"
+            # model = "/Users/zhouchao/Library/CloudStorage/OneDrive-macrosolid/PycharmProjects/wood_color/models/model_2020-11-08_20-49.p"
-            # model = "C:/Users/FEIJINTI/PycharmProjects/wood_color/models/model_2020-11-08_20-49.p"
+            model = r"C:\Users\FEIJINTI\OneDrive\PycharmProjects\wood_color\models\model_2023-03-27_16-32.p"
             model = model.encode('ascii')
             length = len(model) + 4
             length = length.to_bytes(4, byteorder='big')
@@ -156,7 +156,7 @@ def main():
             # else:
             #     print('指令执行失败')
         elif cmd == 'KM':
-            model = "E:/wood_color/data/2023318"
+            model = r"C:\Users\FEIJINTI\OneDrive\PycharmProjects\wood_color\data\316"
             # model = "C:/Users/FEIJINTI/PycharmProjects/wood_color/models/model_2020-11-08_20-49.p"
             model = model.encode('ascii')
             length = len(model) + 4

classifer.py

@@ -238,7 +238,7 @@ class WoodClass(object):
         """
         if file_name is None:
             file_name = "model_" + time.strftime("%Y-%m-%d_%H-%M") + ".p"
-            file_name = os.path.join(ROOT_DIR, "models", file_name)
+        file_name = os.path.join(ROOT_DIR, "models", file_name)
         model_dic = {"n": self.n, "k": self.k, "p1": self.p1, "pur": self.pur, "model": self.model,
                      "ww": self.ww, "hh": self.hh, "width": self.width, "w": self.w, "h": self.h,
                      "mode": self._single_pick, "isCorrect": self.isCorrect, "left_correct": self.left_correct}

socket_detector.py

@@ -35,16 +35,16 @@ def process_cmd(cmd: str, data: any, connected_sock: socket.socket, detector: Wo
     elif cmd == 'TR':
         detector = WoodClass(w=4096, h=1200, n=8000, p1=0.8, debug_mode=False)
         model_name = None
-        if ":" in data:
+        if "$" in data:
-            data, model_name = data.split(":", 1)
+            data, model_name = data.split("$", 1)
             model_name = model_name + ".p"
         settings.data_path = data
         settings.model_path = ROOT_DIR / 'models' / detector.fit_pictures(data_path=settings.data_path, file_name=model_name)
-        response = simple_sock(connected_sock, cmd_type=cmd)
+        response = simple_sock(connected_sock, cmd_type=cmd, result=result)
     elif cmd == 'MD':
         settings.model_path = data
         detector.load(path=settings.model_path)
-        response = simple_sock(connected_sock, cmd_type=cmd)
+        response = simple_sock(connected_sock, cmd_type=cmd, result=result)
     elif cmd == 'KM':
         x_data, y_data, labels, img_names = detector.get_kmeans_data(data, plot_2d=False)
         result = detector.data_adjustments(x_data, y_data, labels, img_names)

utils.py

@@ -280,7 +280,7 @@ def ack_sock(send_sock: socket.socket, cmd_type: str) -> bool:
     return True
 
 
-def done_sock(send_sock: socket.socket, cmd_type: str, result) -> bool:
+def done_sock(send_sock: socket.socket, cmd_type: str, result = '') -> bool:
     '''
     发送任务完成指令
     :param cmd_type:指令类型