在python（PIL）中将另一个图像放在另一个图像的上方，然后将另一个图像隐藏在另一个内部

Question

任务是制作两个功能，一个将灰度图像嵌入到RGB中。第二个显示隐藏的“水印”。我已经完成了任务（有15个测试代码必须通过，我将它们全部传递给D），但是如果可能的话，我想整理一下我的代码。有人告诉我，我可以使用裁剪功能将水印置于照片的上方（我不知道如何操作），而是使用了数学逻辑。

from PIL import Image 
#function for embedding the watermark 
def add_watermark(clean_file, watermark_file): 

#if the files are not compatible 
#or doesn't exist end the function 
try: 
    clean_photo = Image.open(clean_file) 
    watermark_photo = Image.open(watermark_file) 
except: 
    return False 

#the images pixels tuple values are attained 
clean_photo_size = clean_photo.size 
watermark_photo_size = watermark_photo.size 

# .load was used to return a pixel access object 
# that can be used to read and modify pixels 
#for both images 
photo_array = clean_photo.load() 
watermark_array = watermark_photo.load() 

#centring the watermarked image on the photo 
start_x_coord=int((clean_photo_size[0]-watermark_photo_size[0])/2) 
start_y_coord=int((clean_photo_size[1]-watermark_photo_size[1])/2) 

#for the pixels that share the same position as the pixels in the 
#watermark, manipulate their RGB tuple value to include the watermarks 
#greyscale value as part of the original RGB tuple 
for line in range(watermark_photo_size[0]): 
    for rank in range(watermark_photo_size[1]): 
     photo_array [(start_x_coord+line),\ 
       (start_y_coord+rank)]\ 
       =(int(((photo_array [(start_x_coord+line),\ 
            (start_y_coord+rank)][0]/10)*10)\ 
         +((watermark_array[line,rank]/100)%10)),\ 
        int(((photo_array [(start_x_coord+line),\ 
            (start_y_coord+rank)]\ 
         [1]/10)*10)+((watermark_array[line,rank]/10)%10)),\ 
        int(((photo_array [(start_x_coord+line),\ 
            (start_y_coord+rank)][2]/10)*10)\ 
         +(watermark_array[line,rank]%10))) 

#create a new file name with _X at the end 
new_File_Name = clean_file[:-4]+"_X"+clean_file[-4:] 

#create a new file 
clean_photo.save(new_File_Name) 

#return true for the test 
return True 

我试图让这个职位的拼写和语法是正确的，但我一直清醒20小时，所以我现在很抱歉，如果我错过了什么。

Answer 1

这里是你如何可以在另一个图像的顶部中央的图像水印：

注：图片和ImageEnhance要求PIL系统库安装到Python运行时

import os 
from os.path import join, exists 
import shutil 
import Image 

class BatchWatermark: 

    def __init__(self): 
     self.data = [] 


    def process(self, infolder, outfolder, watermarkImage): 
     if not exists(infolder): 
      print("Input folder '" + infolder + "' does not exist") 
      raise Exception("Input folder '" + infolder + "' does not exist") 
     #if exists(outfolder): 
     # shutil.rmtree(outfolder) 

     watermark = Image.open(watermarkImage) 
     for root, dirs, files in os.walk(infolder): 
      for name in files:  
       try: 
        im = Image.open(join(root, name)) 
        if im.mode != 'RGBA': 
         im = im.convert('RGBA') 
         layer = Image.new('RGBA', im.size, (0, 0, 0, 0)) 

         wm_x, wm_y = watermark.size[0], watermark.size[1] 
         x = (im.size[0] - wm_x)/2 
         y = (im.size[1] - wm_y)/2 
         position = (x, y) 
         layer.paste(watermark, position) 
         if not exists(outfolder): 
          os.makedirs(outfolder) 
         Image.composite(layer, im, layer).save(join(outfolder, name)) 
         print "Watermark added to image file '" + outfolder + "/" + name + "'"; 
       except Exception, (msg): 
        print msg