我可以将 TensorFlow Object Detection API 的输出图像保存在文件夹中吗？答案

【问题标题】：Can I save the output images of TensorFlow Objectdetection API in a folder?我可以将 TensorFlow Object Detection API 的输出图像保存在文件夹中吗？
【发布时间】：2019-06-25 14:49:41
【问题描述】：

我是对象检测的新手。在 Jupyter notebook 中检测完成后，我可以将我的 tensorflow 对象检测 API 的输出图像保存在文件夹中吗？

在检测到几张图像时右键单击并保存图像很直观，但当我有超过 1000 张图像要运行检测时，这样做是不切实际的。

输出是否已存储在我不知道的子文件夹中？我可以将输出存储在文件夹或目录中吗？

任何建议将不胜感激。我检查了不同的方法，但没有发现任何帮助

这就是我的检测单元的样子谢谢

【问题讨论】：

标签： tensorflow jupyter-notebook object-detection object-detection-api

【解决方案1】：

使用 PIL，您可以将图像保存到磁盘。首先你导入模块

from PIL import Image

然后你可以注释掉情节线，添加保存线。

#plt.figure(figsize=IMAGE_SIZE)
#plt.imshow(image_np)
#save to same folder as data input
img = Image.fromarray(image_np)
img.save(image_path[:-4]+'output_'+'.png')

【讨论】：

很遗憾地感谢@danyfang，我仍然收到错误ValueError Traceback（最近一次调用最后一次）---> 36 img.save（save_path）/home/v/.local/lib/python2 .7/site-packages/PIL/Image.pyc in save(self, fp, format, **params) 1987 format = EXTENSION[ext] 1988 除了 KeyError: -> 1989 raise ValueError('unknown file extension: {}' .format(ext)) 1990 1991 如果 format.upper() 不在 SAVE 中：ValueError: unknown file extension:'
我已经更新了代码，你可以重试我的解决方案。
感谢@danyfang 的及时回复。扩展错误消失了，但我看不到我的输出.....它们存储的新路径在哪里，因为我没有将它指向我选择的目录。再次感谢。 PS：我使用的是github仓库tensorflow/object_deteciton：github.com/tensorflow/models/tree/master/research/….
输出保存在与输入图像相同的路径，并添加了'output_'
在../models/research/object_detection/test_images/下