写了个批量为视频添加文字水印和图片水印的程序

ClearLover

如题，代码如下，可设置多个图片水印及它们的移动位置

上面的介绍有点太敷衍了，重新写，
功能为：可以添加多个动态移动的水印，还可以设置水印的大小以及移动速度，也可以增加文字水印，重点是这个是批量执行的，可以对目录下的所有视频文件批量添加水印

[Python] 纯文本查看 复制代码

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[i]
            watermark_size = watermark_sizes[i]
            watermark_speed = watermark_speeds[i]

            # 读取水印图片
            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[1])
                    y = np.random.randint(0, height - watermark_image.shape[0])
                    too_close = False

                    # 检查新位置与已选择水印位置之间的距离
                    for j in range(i):
                        dist = np.sqrt((x - watermark_positions[j][0]) ** 2 + (y - watermark_positions[j][1]) ** 2)
                        if dist < watermark_image.shape[1] or dist < watermark_image.shape[0]:
                            too_close = True
                            break

                    if not too_close:
                        watermark_positions[i] = (x, y)
                        break
            else:
                x, y = watermark_positions[i]

            # 在帧上添加水印
            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[y : y + watermark_resized.shape[0], x : x + watermark_resized.shape[1], c] = (
                    frame[y : y + watermark_resized.shape[0], x : x + watermark_resized.shape[1], 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)

很多人都在回复说要成品，虽然我不知道为什么大家都觉得保存一下很困难，但是我还是应大家要求保存了一个，自己下载吧.....（不会真的有人还来下载吧？？？）
批量视频加水印.zip (2.25 KB, 下载次数: 269)

2023-7-13 18:20 上传

点击文件名下载附件
里面就是个py文件而已
下载积分: 吾爱币 -1 CB

ClearLover

zoomyou 发表于 2023-7-13 16:28
这种加水印的软件，市面多的是，有本事开发批量去除文件狗皮膏药的、去除水印的、尤其是去除图片一体水印的 ...

求分享两个，我就是因为找不到才只好自己写的，好人一生平安

ClearLover

gzhpsz 发表于 2023-8-28 11:09
faststone capture 只能一张一张地添加文字，我想偷懒，要快速方便地把整个文件中的照片，一健添加文字，而 ...

图片加水印的批量Python程序实际上我写了两个版本，一个版本是加图片水印的，可以控制图片水印大小和位置，默认是左上右上左下居中，还有个加文字水印的，本来想二合一后发出来的，现在不想发了，奥，对了，但这两个会在加水印后压缩图片，当然，也有一个版本是加水印后无损甚至会尝试修复画质的，不过提升不大，比如22MB的写真图片，加水印提升质量后到25MB左右

ClearLover

对了，有的人可能不会安装库，顺便打在下面了
pip install Pillow

bihui

可以可以，厉害

jzzh

程序呢哥

ClearLover

bihui 发表于 2023-7-13 15:24
可以可以，厉害

来点爱心如何

ClearLover

jzzh 发表于 2023-7-13 15:30
程序呢哥

不是在上面

wangwh

虽然看不懂，仍然给个好评

lye123456

再加的功能。可以去水印的，更完美。谢谢分享

ClearLover

lye123456 发表于 2023-7-13 15:54
再加的功能。可以去水印的，更完美。谢谢分享

试过了，去水印的.....由于只能调用本地库，没有很好的AI能力，所以结果太一言难尽了，说去水印吧，它确实去除了，但是还不如不去，就是一团花

chayunyuxiang

只有代码，没有集成的程序吗？