我的问题很简单,如何在 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