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

Question

I am new to object detection. Can I save the output images of my tensorflow object detection API in a folder after detection is completed in the Jupyter notebook?

It was intuitive to right click and save the images when detecting a few images, but it is impractical to do same when I have over 1000 images to run detection on.

is the output already stored in a subfolder that I am unaware of?? Can I store the output in a folder or directory?

Any suggestions will be appreciated. I have checked differnt means but found nothing helpful

this is what my detection cell looks like Thanks

Answer 1

Use PIL you can save your image to disk. First you import the module

from PIL import Image

And then you can comment out the plot lines, add the save lines.

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