【Python入门教程】OpenCV在图片/视频上添加Logo（水印）
还是老样子，最近项目需要在视频上添加logo，所以就找了一下相关资料，然后写了一段代码，今天给大家分享一下如何使用Python的OpenCV库给图片或视频添加水印和logo。
一、导入库
OpenCV库导入的时候是cv2
import cv2二、代码部分
2.1 基础逻辑
- 通过二值化阈值筛选出需要的Logo部分
- 将logo部分制作成掩膜
- 使用掩膜和logo范围提取图片的背景
- 使用反向掩膜提取需要的logo部分
- 将背景和logo叠加显示到原图上
2.2 完整代码
这里我做了三个入参，图片地址、logo地址、logo放缩的尺寸。logo放在左上角，尺寸视情况自己修改。
# -*- coding: utf-8 -*-
"""
@Time ： 2023/10/20 17:34
@Auth ： RS迷途小书童
@File ：Image_Add_Logo.py
@IDE ：PyCharm
@Purpose：图片/视频添加水印
"""
import cv2
def Image_add_logo(image_path, logo_path, size=(403, 161)):
image1 = cv2.imread(image_path)
image2 = cv2.imread(logo_path)
image2 = cv2.resize(image2, size)
rows, cols, channels = image2.shape
# 获取logo的行列、通道数
roi = image1[0:rows, 0:cols]
# 从image1中切割出一个与image2大小相同的区域，作为ROI（Region of Interest)
image2gray = cv2.cvtColor(image2, cv2.COLOR_BGR2GRAY)
# 将image2转化为灰度图像，以便于进行阈值化处理
ret, mask = cv2.threshold(image2gray, 254, 255, cv2.THRESH_BINARY)
# ret阈值被忽略，mask二值化图像。大于254的值赋值为255，小于254的值赋值为0
mask_reverse = cv2.bitwise_not(mask)
# 按位取反操作，0变255，255变0
image1_background = cv2.bitwise_and(roi, roi, mask=mask)
# 按位与操作,保留掩膜之后的背景
image2_logo = cv2.bitwise_and(image2, image2, mask=mask_reverse)
# 按位与操作,保留掩膜之后的logo
result = cv2.add(image1_background, image2_logo)
# 将背景和logo叠加
image1[0:rows, 0:cols] = result
# 将新的图像覆盖到原图像image1的相应位置
# cv2.imshow("Image", image1) # 显示叠加图像
# cv2.waitKey(0)
cv2.destroyAllWindows()
return image1
if __name__ == "__main__":
Image = r"G:\Neat Download Manager\Compressed\video/100000022.png"
Logo = r"G:\Neat Download Manager\Compressed\video/logo.png"
Size = (604, 242)
image = Image_add_logo(Image, Logo, Size)
三、总结
这个博文没有写如何在视频中添加logo，因为原理是一样的。cv2库在读取视频时就是逐帧读取，所以本质就是图片，我之前已经发过如何使用OpenCV库将视频拆帧成图片，然后如何用图片生成视频，所以这里就不分享了。只要大家将这两篇博文的内容组合一下，就可以实现视频添加Logo，链接：基于OpenCV视频分解成图片+图片组合成视频（视频抽帧组帧）