如何在 Keras 中找到与预测关联的文件名?

数据挖掘 Python 喀拉斯
2022-02-12 20:22:16

我的问题很简单,如何在 Keras 中找到与预测相关的文件名?也就是说,如果我有一组 100 个名为的测试样本,并且我得到一个包含估计的类概率的 numpy 数组,我如何将文件名映射到概率?

import cv2
import os
import glob 

def load_test():
    X_test = []
    y_test = []
    os.chdir(testing_path)
    file_list = glob.glob('*.png')
    for test_image in file_list:
        img = cv2.imread(test_image,1)
        X_test.append(img)
        y_test.append(1)
   return X_test,y_test

if __name__ == '__main__':
   X_test = np.array(X_test, dtype = np.uint8)
   X_test = X_test.reshape(X_test.shape[0],3,100,100)
   X_test = X_test.astype('float32')
   X_test /= 255
1个回答

填充 file_list 的文件的顺序与 X_test 按行出现的顺序相同。

所以只需匹配索引以将文件名与预测相关联。

X_test[0] ~ 预测[0] ~ file_list[0]