1. 程式人生 > >python將將影象矩陣轉換為向量

python將將影象矩陣轉換為向量

假設影象矩陣大小為32×32,將其轉換為向量,首先建立1×1024的NumPy陣列,然後開啟給定的檔案,迴圈讀出檔案的前32行,並將每行的頭32個字元值儲存在NumPy陣列中

import numpy as np

def img2vector(filename):
    returnVect = np.zeros((1, 1024))
    fr = open(filename)
    for i in range(32):
        lineStr = fr.readline()
        for j in range(32):
            returnVect[0, 32*i+j] = int(lineStr[j])
    return returnVect