diff --git a/function.py b/function.py
index 9347324..6c71411 100644
--- a/function.py
+++ b/function.py
@@ -73,7 +73,9 @@ def img_to_vector(filename):
         lines = file.readlines()
         n = 0
         for line in lines:
-            for i in range(0, len(line)):
+            # 去除每行末尾的换行符
+            line = line.strip()
+            for i in range(len(line)):
                 returnVect[0, n] = int(line[i])
                 n = n + 1
                 if n >= 1024: