更新 function.py

function.py

@@ -83,7 +83,12 @@ def img_to_vector(filename):
     - np.array: 二值图像向量。
     """
     with open(filename, 'r') as file:
-        return np.array([int(char) for char in file.read()]).reshape(-1, 1)
+        lines = file.readlines()
+    returnVect = np.zeros((1, 1024))
+    for i in range(32):  # 假设每行有32个字符
+        for j in range(32):  # 假设有32行
+            returnVect[0, 32 * i + j] = int(lines[i][j])
+    return returnVect
 
 if __name__ == "__main__":
     train_data, train_labels = load_dataset("trainingDigits")