How to save an image from a custom COCO dataset with its annotations overlaid on top of it

Question

I created a custom COCO dataset. Now suppose I have valid image metadata in image_data. I can display the image and the annotation with

import skimage.io as io
import matplotlib.pyplot as plt

image_directory ='my_images/'

image = io.imread(image_directory + image_data['file_name'])
plt.imshow(image); plt.axis('off')
pylab.rcParams['figure.figsize'] = (8.0, 10.0)
annotation_ids = example_coco.getAnnIds(imgIds=image_data['id'], catIds=category_ids, iscrowd=None)
annotations = example_coco.loadAnns(annotation_ids)
example_coco.showAnns(annotations)

At this point, I'll be able to see annotations overlaid over the image. However, I want to save the image with the annotations overlaid on top of it. How can I do that? I tried

io.imsave(fname="test.png", arr=image)

But it doesn't work. It just saves the original image, without any annotation.

annotation is not draw directly on image but on screen. You could use PIL/pillow to draw directly on image. [Draw a rectangle and a text in it using PIL](https://stackoverflow.com/questions/41405632/draw-a-rectangle-and-a-text-in-it-using-pil/41406147#41406147) — furas, Sep 04 '19 at 15:45

furas · Answer 1 · 2019-09-04T16:32:05.063

You can save figure/plot using

 plt.savefig("test.png", bbox_inches='tight', pad_inches=0)

There are some parameters to remove margins.

Working example

import skimage.io as io
import matplotlib.pyplot as plt

image = io.imread("https://homepages.cae.wisc.edu/~ece533/images/lena.png")

plt.imshow(image)
plt.axis('off')
plt.annotate("Lena", (10, 20))
plt.savefig("test.png", bbox_inches='tight', pad_inches=0) # if you want to display then `savefig()` has to be before `show()` 
#plt.show()

Eventually you can use PIL/pillow to draw directly on image

import skimage.io as io
from PIL import Image, ImageDraw, ImageFont

image = io.imread("https://homepages.cae.wisc.edu/~ece533/images/lena.png")

img = Image.fromarray(image)
draw = ImageDraw.Draw(img)
draw.text((10,10), "Lena", font=ImageFont.truetype("arial", 20), fill=(0,0,0))
img.save('test.png')

How to save an image from a custom COCO dataset with its annotations overlaid on top of it

1 Answers1