How can I add an image watermark to only those images which are to be published in gallery using a function in functions.php? Images can be in .jpg, .png, .gif format.

I also found this snippet which serves the same but it watermarks all the uploaded images.