util.py 文件源码

def get_figs(dir_name):
    ret = []
    for file_name in os.listdir(dir_name):
        #tmp = cv2.imread(os.path.join(dir_name, file_name), cv2.IMREAD_GRAYSCALE)
        #tmp = np.reshape(tmp, (64, 64, 1))
        tmp = cv2.imread(os.path.join(dir_name, file_name))
        ret.append(tmp/127.5 - 1.0)
    return np.asarray(ret, dtype = np.float32)