写了个批量为视频添加文字水印和图片水印的程序
本帖最后由 ClearLover 于 2023-7-13 18:20 编辑如题，代码如下，可设置多个图片水印及它们的移动位置
上面的介绍有点太敷衍了，重新写，
功能为：可以添加多个动态移动的水印，还可以设置水印的大小以及移动速度，也可以增加文字水印，重点是这个是批量执行的，可以对目录下的所有视频文件批量添加水印
import cv2
import os
import numpy as np
from moviepy.editor import VideoFileClip
def add_dynamic_watermarks_with_audio(video_path, output_path, watermark_image_paths, watermark_sizes, watermark_speeds, add_text_watermark, text_watermark_text, text_watermark_color):
# 读取视频
video_clip = VideoFileClip(video_path)
audio_clip = video_clip.audio
# 获取视频的宽度和高度
width, height = video_clip.size
# 准备文本水印
if add_text_watermark:
# 创建用于绘制文本水印的画布
text_watermark_canvas = np.zeros((height, width, 3), dtype=np.uint8)
# 设置文本水印属性
font_face = cv2.FONT_HERSHEY_SIMPLEX
font_scale = 1.5
font_thickness = 2
# 获取文本水印的大小
(text_width, text_height), _ = cv2.getTextSize(text_watermark_text, font_face, font_scale, font_thickness)
# 计算文本水印的位置
text_x = int((width - text_width) / 2)
text_y = int(height - text_height - 10)
# 在画布上绘制文本水印
cv2.putText(text_watermark_canvas, text_watermark_text, (text_x, text_y), font_face, font_scale, text_watermark_color, font_thickness, cv2.LINE_AA)
# 在每一帧上添加水印
watermark_positions = [(0, 0)] * len(watermark_image_paths)
def process_frame(t, x):
nonlocal watermark_positions
frame = video_clip.get_frame(t)
# 在帧上添加图片水印
for i in range(len(watermark_image_paths)):
watermark_image_path = watermark_image_paths
watermark_size = watermark_sizes
watermark_speed = watermark_speeds
# 读取水印图片
watermark_image = cv2.imread(watermark_image_path)
# 调整水印图片大小
watermark_height, watermark_width, _ = watermark_image.shape
if watermark_width > width or watermark_height > height:
scale_factor = min(width / watermark_width, height / watermark_height)
watermark_image = cv2.resize(
watermark_image,
(int(watermark_width * scale_factor), int(watermark_height * scale_factor)),
interpolation=cv2.INTER_LINEAR,
)
# 更新水印位置
if t % watermark_speed == 0:
while True:
x = np.random.randint(0, width - watermark_image.shape)
y = np.random.randint(0, height - watermark_image.shape)
too_close = False
# 检查新位置与已选择水印位置之间的距离
for j in range(i):
dist = np.sqrt((x - watermark_positions) ** 2 + (y - watermark_positions) ** 2)
if dist < watermark_image.shape or dist < watermark_image.shape:
too_close = True
break
if not too_close:
watermark_positions = (x, y)
break
else:
x, y = watermark_positions
# 在帧上添加水印
watermark_resized = cv2.resize(
watermark_image, (int(watermark_size * watermark_width), int(watermark_size * watermark_height))
)
alpha = watermark_resized[:, :, 0] / 255.0
for c in range(3):
frame_copy = frame.copy()
frame_copy, x : x + watermark_resized.shape, c] = (
frame, x : x + watermark_resized.shape, c] * (1 - alpha)
+ watermark_resized[:, :, c] * alpha
)
# 在帧上添加文本水印
if add_text_watermark:
frame_with_text = cv2.addWeighted(frame, 1, text_watermark_canvas, 0.7, 0)
frame = frame_with_text
cv2.imshow("Watermarked Video", frame)
cv2.waitKey(1)
return frame
# 处理视频每一帧
processed_clip = video_clip.fl(lambda gf, t: process_frame(t, gf))
# 输出处理后的视频
final_clip = processed_clip.set_audio(audio_clip)
final_clip.write_videofile(output_path, codec='libx264', audio_codec="aac")
return "水印添加完成！"
def batch_add_watermarks_in_directory_with_audio(directory, output_directory, watermark_image_paths, watermark_sizes, watermark_speeds, add_text_watermark=False, text_watermark_text=None):
# 获取目录中的所有文件
file_list = os.listdir(directory)
# 遍历文件列表
for file_name in file_list:
# 检查文件是否为视频文件
if file_name.lower().endswith(('.avi', '.mp4', '.mov', '.mkv')):
file_path = os.path.join(directory, file_name)
# 设置输出文件路径
output_file_path = os.path.join(output_directory, file_name)
# 添加水印
add_dynamic_watermarks_with_audio(file_path, output_file_path, watermark_image_paths, watermark_sizes, watermark_speeds, add_text_watermark, text_watermark_text if add_text_watermark else None, (255, 255, 255))
return "批量处理完成！"
# 使用示例
if __name__ == '__main__':
directory = input("请输入视频文件所在目录路径：")# 视频文件所在目录
output_directory = input("请输入输出视频文件目录路径：")# 输出视频文件目录
# 检查并创建输出目录
os.makedirs(output_directory, exist_ok=True)
watermark_image_paths = []
watermark_sizes = []
watermark_speeds = []
num_watermarks = int(input("请输入要添加的水印数量："))
for i in range(num_watermarks):
watermark_image_path = input(f"请输入第{i+1}个水印图片的路径：")
watermark_size = float(input(f"请输入第{i+1}个水印图片的大小（0~1之间）："))
watermark_speed = int(input(f"请输入第{i+1}个水印图片的速度："))
watermark_image_paths.append(watermark_image_path)
watermark_sizes.append(watermark_size)
watermark_speeds.append(watermark_speed)
add_text_watermark_input = input("是否添加文本水印？(y/n) ")
if add_text_watermark_input.lower() == 'y':
add_text_watermark = True
text_watermark_text = input("请输入文本水印内容：")
else:
add_text_watermark = False
text_watermark_text = None
batch_add_watermarks_in_directory_with_audio(directory, output_directory, watermark_image_paths, watermark_sizes, watermark_speeds, add_text_watermark, text_watermark_text)
很多人都在回复说要成品，虽然我不知道为什么大家都觉得保存一下很困难，但是我还是应大家要求保存了一个，自己下载吧.....（不会真的有人还来下载吧？？？）
zoomyou 发表于 2023-7-13 16:28
这种加水印的软件，市面多的是，有本事开发批量去除文件狗皮膏药的、去除水印的、尤其是去除图片一体水印的 ...
求分享两个，我就是因为找不到才只好自己写的，好人一生平安 gzhpsz 发表于 2023-8-28 11:09
faststone capture 只能一张一张地添加文字，我想偷懒，要快速方便地把整个文件中的照片，一健添加文字，而 ...
图片加水印的批量Python程序实际上我写了两个版本，一个版本是加图片水印的，可以控制图片水印大小和位置，默认是左上右上左下居中，还有个加文字水印的，本来想二合一后发出来的，现在不想发了，奥，对了，但这两个会在加水印后压缩图片，当然，也有一个版本是加水印后无损甚至会尝试修复画质的，不过提升不大，比如22MB的写真图片，加水印提升质量后到25MB左右 对了，有的人可能不会安装库，顺便打在下面了
pip install Pillow 可以可以，厉害 程序呢哥 bihui 发表于 2023-7-13 15:24
可以可以，厉害
来点爱心如何 jzzh 发表于 2023-7-13 15:30
程序呢哥
不是在上面 虽然看不懂，仍然给个好评:lol 再加的功能。可以去水印的，更完美。谢谢分享 lye123456 发表于 2023-7-13 15:54
再加的功能。可以去水印的，更完美。谢谢分享
试过了，去水印的.....由于只能调用本地库，没有很好的AI能力，所以结果太一言难尽了，说去水印吧，它确实去除了，但是还不如不去，就是一团花 只有代码，没有集成的程序吗？