python_给视频打马赛克

python_给视频打马赛克

背景说明:据我了解，视频处理常用的是moviepy，而cv2更多则是用于图像处理，音频需单独处理；本文就是将两者进行结合使用，实现对视频打马赛克。众所周知，使用moviepy对视频进行打马时，大多数的教程都是用 mp.ImageClip 或者 mp.ColorClip 给视频加一个遮罩处理的，效果并不理想(如下图);

实现打马后效果展示

代码说明

import moviepy.editor as mp
import cv2

#对视频的某一帧进行打码，并返回打码后的图片
def do_mosaic(frame, x, y, w, h, neighbor=5):
    """
    马赛克原理: 将图像上某个像素点一定邻域内的所有点用邻域内左上像素点的颜色代替，这样模糊细节的同时可以保留大致的轮廓。（也可以考虑将每一帧的图片指定区域的像素点进行随机打乱处理）
    :param frame: opencv frame
    :param int x:  马赛克左顶点x值(这几个参数可以参考下面的示意图) 
    :param int y:  马赛克左顶点y值
    :param int w:  马赛克的宽度值
    :param int h:  马赛克的高度值
    :param int neighbor:  马赛克每一块的宽
    """
    for i in range(0, h - neighbor, neighbor):
        for j in range(0, w - neighbor, neighbor):
            rect = [j + x, i + y]

			# 其中一个点的RGB值
            color = frame[rect[1]][rect[0]].tolist()
            
            left_up = (rect[0], rect[1])
            right_down = (rect[0] + neighbor - 1, rect[1] + neighbor - 1)  
            
            cv2.rectangle(frame, left_up, right_down, color, -1)
    return frame

# 处理视频中的每一帧图片
def scroll(get_frame, t):
    # 获取每一帧图片
    frame = get_frame(t)
    
	# 提取视频的宽和高
    fh, fw = frame.shape[0], frame.shape[1]
    
    # 马赛克的区域（视频的右上角） fh:视频的高度，fw视频的宽度（可以参考下面示意图）
    x = int(fw * 0.75)
    y = 0
    w = int(fh * 0.25)
    h = int(fw * 0.1)

    frame_region = do_mosaic(frame, x, y, w, h, neighbor=5)
    return frame_region

# 视频处理
def video_processing(path_src, path_save):
    '''
    :param path_src:
    :param path_save:
    :return:
    '''
    # 读取视频
    video = mp.VideoFileClip(path_src).subclip(5, -5)  # 减去视频的前后各5s

    # 对每一帧进行打码,
    video_new = video.fl(scroll)

    # 视频导出
    video_new.write_videofile(path_save)
    print("导出成功")
    # 释放内存
    video_new.close()


if __name__ == '__main__':
    # 处理前的视频文件路径
    path1 = './src.mp4'
    # 处理后保存的文件路径
    path2 = './output.mp4'

    # 处理视频
    video_processing(path1, path2)

示意图