实现对图片或者视频增长隐蔽水印和提取水印

好久好久没有写博客了，最近看见一个很故意思的文章：鉴戒你的电脑被窃听，就是说在一些公司，截图都会存在水印，方便溯源，然后出于技术的好奇，我在github上搜了一下，还真有相干的github项目：对图片举行水印增长和提取，下面是地址：blind_watermark
1.实现对图片的水印提取
pip install blind-watermark
# 嵌入水印： blind_watermark --embed --pwd 1234 examples/pic/ori_img.jpeg "watermark text" examples/output/embedded.png
# 提取水印： blind_watermark --extract --pwd 1234 --wm_shape 111 examples/output/embedded.png 
很简单，就是按着命令后走就行，但是问题是：公司的图片里面隐蔽的水印的加密方式，我们也不知道啊，我提取了一下我的截图，出现的就是乱码
2.实现对视频的水印添加和提取
这个没什么，就是在上面的基础上，对视频举行分帧，然后对每帧举行水印的添加，然后再加添加了水印的图片合成video，另有对每帧举行水印的提取。

1. import os
2. from blind_watermark import WaterMark
3. import cv2
4. def frames_to_video(frame_folder, output_video_path, frame_rate):
5.     frame_paths = sorted([os.path.join(frame_folder, f) for f in os.listdir(frame_folder)])
6.     first_frame = cv2.imread(frame_paths[0])
7.     height, width, _ = first_frame.shape
9.     fourcc = cv2.VideoWriter_fourcc(*'mp4v')
10.     video = cv2.VideoWriter(output_video_path, fourcc, frame_rate, (width, height))
12.     for frame_path in frame_paths:
13.         frame = cv2.imread(frame_path)
14.         video.write(frame)
16.     video.release()
18. def embed_watermark_on_frames(input_folder, output_folder, password_img, password_wm, watermark):
19.     bwm = WaterMark(password_img=password_img, password_wm=password_wm)
20.     bwm.read_wm(watermark, mode='str')
22.     for frame_path in os.listdir(input_folder):
23.         frame_full_path = os.path.join(input_folder, frame_path)
24.         bwm.read_img(frame_full_path)
25.         output_frame_path = os.path.join(output_folder, frame_path)
26.         bwm.embed(output_frame_path)
28. def video_to_frames(video_path, output_folder):
29.     cap = cv2.VideoCapture(video_path)
30.     frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
31.     frame_rate = int(cap.get(cv2.CAP_PROP_FPS))
33.     frame_idx = 0
34.     while cap.isOpened():
35.         ret, frame = cap.read()
36.         if not ret:
37.             break
38.         frame_path = f"{output_folder}/frame_{frame_idx:04d}.png"
39.         cv2.imwrite(frame_path, frame)
40.         frame_idx += 1
42.     cap.release()
43.     return frame_count, frame_rate
44. def extract_watermark_from_frames(frame_folder, password_img, password_wm, wm_shape):
45.     bwm = WaterMark(password_img=password_img, password_wm=password_wm)
46.     for frame_path in os.listdir(frame_folder):
47.         frame_full_path = os.path.join(frame_folder, frame_path)
48.         watermark = bwm.extract(frame_full_path, wm_shape=wm_shape, mode='str')
49.         print(f"Watermark from {frame_path}: {watermark}")
51. # 分解视频
52. video_path = "test.mp4"
53. output_frames_folder = "frames"
54. os.makedirs(output_frames_folder, exist_ok=True)
55. frame_count, frame_rate = video_to_frames(video_path, output_frames_folder)
57. # 嵌入水印
58. watermark = "secret"
59. password_img = 1
60. password_wm = 1
61. output_watermarked_frames = "watermarked_frames"
62. os.makedirs(output_watermarked_frames, exist_ok=True)
63. embed_watermark_on_frames(output_frames_folder, output_watermarked_frames, password_img, password_wm, watermark)
65. # 合成为视频
66. output_video_path = "watermarked_video.mp4"
67. frames_to_video(output_watermarked_frames, output_video_path, frame_rate)
69. # 提取水印
70. extract_watermark_from_frames(output_watermarked_frames, password_img, password_wm, wm_shape=len(watermark))

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。