时间戳

首页 > 解决方案 > 我想在预测后将图像列表写入文件夹。我该如何解决？

python - 我想在预测后将图像列表写入文件夹。我该如何解决？

问题描述

predict=[]
for img in glob.glob("test1/*.jpg"):
    n= cv2.imread(img)
    predicted = model.prediction(n)
    predict.append(predicted)

如何将此预测图像列表作为图像保存到文件夹中？模型返回 224x224 灰度图片。

标签： pythondeep-learningconv-neural-networkprediction

解决方案

对列表中的每个图像使用imwrite ：

cv2.imwrite(filename, predicted)

这会将名称为filename的图像保存在当前工作目录中。如果要保存到另一个文件夹，请输入文件的路径而不是文件名：

cv2.imwrite(path_to_file, predicted)

推荐阅读