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 = []