Why not replace drawings_object.writeImage with flag color mapping?

[dogod621]

The code of drawings_object.writeImage is weird...

why not use:

def writeImage(image):
    """ store label data to colored image """
    plt.imshow(image, cmap=plt.get_cmap('flag'))
    

it is much simpler and without a fixed class types.
It works fine to me.