From 24ca1099e6d8db3f0599f6e6c7a245070fb9a07a Mon Sep 17 00:00:00 2001
From: duanmu <774052669@qq.com>
Date: Thu, 30 Mar 2023 16:43:52 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E6=94=B9=E4=BA=86SZQ=E5=A4=A7?=
 =?UTF-8?q?=E5=B0=8F=E5=86=99?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 QT_test.py   | 1 +
 classifer.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/QT_test.py b/QT_test.py
index 27a475e..1514b17 100644
--- a/QT_test.py
+++ b/QT_test.py
@@ -165,6 +165,7 @@ def main():
             socket_send_1.send(send_message)
             print('发送成功')
             result = socket_send_2.recv(5)
+            print(result)
             new_leng = int.from_bytes(result[1:], byteorder='big')
             result = socket_send_2.recv(new_leng)
             print(result)
diff --git a/classifer.py b/classifer.py
index 521a963..204df11 100644
--- a/classifer.py
+++ b/classifer.py
@@ -507,7 +507,7 @@ class WoodClass(object):
         y_data = y_data[sorted_idx]
         labels = labels[sorted_idx]
         img_names = [img_names[i] for i in sorted_idx]
-        mapping = {0: 's', 1: 'z', 2: 'q'}
+        mapping = {0: 'S', 1: 'Z', 2: 'Q'}
         y_data = [mapping[i] for i in y_data]
         labels = [mapping[i] for i in labels]
         data = []