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

Question

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

当检测到一些图像时，右键单击并保存图像是很直观的，但是当我有1000多个图像需要进行检测时，这样做是不切实际的。

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

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

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

Answer 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')