给图片批量添加水印

2023-12-02

使用OpenCV给图片批量添加水印

导言：平时摄影较多，于是开始想使用python批量给照片添加水印

addWeighted函数


src1	first input array.
alpha	weight of the first array elements.
src2	second input array of the same size and channel number as src1.
beta	weight of the second array elements.
gamma	scalar added to each sum.
dst	output array that has the same size and number of channels as the input arrays.
dtype	optional depth of the output array; when both input arrays have the same depth, dtype can be set to -1, which will be equivalent to src1.depth().

背景为黑色时，值为0不受权重值影响

import os
import cv2
import numpy as np
from PIL import Image
def findAllFiles(root_dir,filter):
    print("find files end with \'" + filter+"...")
    paths=[]
    names=[]
    files=[]
    seperator=os.path.sep
    for parent,dirnames,filenames in os.walk(root_dir):
        for filename in filenames:
            if filename.endswith(filter):
                paths.append(parent+seperator)
                names.append(filename)

    for i in range(paths.__len__()):
        files.append(paths[i]+names[i])
        print(files)
    paths.sort()
    names.sort()
    files.sort()
    return paths,names,files

def addLogo(img_lists,logo_dir,save_path="C:\\Users\\11062\\Desktop\\pictures\\withlogo"):
    logo_img=cv2.imread(logo_dir)
    parts=[]
    for item in img_lists:
        width=item.shape[1]
        height=item.shape[0]
        #new_logo_width=int(width/5)
        logo_img_new=cv2.resize(logo_img,(int(logo_img.shape[1]),int(logo_img.shape[0])),interpolation=cv2.INTER_AREA)
        logo_img_new_dim=(int(logo_img.shape[1]),int(logo_img.shape[0]))
        center_y=int(height/2)
        center_x=int(width/2)
        top_y=center_y-int(logo_img.shape[0]/2)
        left_x=center_x-int(logo_img.shape[1]/2)
        bottom_y=top_y+logo_img_new.shape[0]
        right_x=left_x+logo_img_new.shape[1]
        roi=item[top_y:bottom_y,left_x:right_x]
        result=cv2.addWeighted(roi,1,logo_img_new,0.3,0)
        item[top_y:bottom_y,left_x:right_x]=result
        #parts.append(item)
        image_data=np.asarray(item)
        parts.append(image_data)
    for i in range(len(parts)):
        cv2.imwrite(save_path+"/part_"+i.__str__()+".jpg",parts[i])

if __name__=='__main__':
    parts=[]
    _,_,imgs=findAllFiles("C:\\Users\\11062\\Desktop\\pictures",".jpg")
    #print(imgs)
    for item in imgs:
        parts.append(cv2.imread(item))

    addLogo(parts,"C:\\Users\\11062\\Desktop\\pictures\\kao.jpg","C:\\Users\\11062\\Desktop\\pictures\\withlogo")

参考：

OpenCV: Operations on arrays

[如何在 Python 中使用 OpenCV 在图像上添加水印-CSDN博客](https://blog.csdn.net/woshicver/article/details/120715078?ops_request_misc=%7B%22request%5Fid%22%3A%22170152622116800215090835%22%2C%22scm%22%3A%2220140713.130102334.pc%5Fall.%22%7D&request_id=170152622116800215090835&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~all~first_rank_ecpm_v1~rank_v31_ecpm-11-120715078-null-null.142^v96^pc_search_result_base7&utm_term=opencv给图片添加水印 python&spm=1018.2226.3001.4187)