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

Can I save the output images of TensorFlow Objectdetection API in a folder?

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

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

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

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

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

使用 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')