python 给视频添加马赛克

用法：

1. 创建空文件夹：imgs

2. 将倒数第三行中的"222056.mov"改为你的视频路径，如："a.mov"

3. 运行以下代码

4. 稍等片刻，鼠标拖动选择添加马赛克区域，按enter确定（每次操作只能添加一个马赛克区域，视频会根据所选区域全程添加马赛克）

5. 稍等片刻，'a.avi '就是做好的视频文件，！！！注意！！！该视频文件没有音频

展望：

可以结合opencv的图像修复功能实现消除图标，效果比一坨马赛克好上不知几倍！有空再搞……

源码：

import cv2, os

import numpy as np

def video2img(videoroot):

    cap = cv2.VideoCapture(videoroot)

    isOpened = cap.isOpened    # 判断是否打开‘

    print(isOpened)

    fps = cap.get(cv2.CAP_PROP_FPS)    #帧率

    width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))

    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))

    print(fps,width,height)

    i = 0

    while(isOpened):

        i += 1

        (flag,frame) = cap.read() # 读取每一张 flag frame

        fileName = './imgs/image'+str(i)+'.jpg'

        print(fileName)

        if flag == True:

            frame = np.rot90(frame, 1)

            cv2.imwrite(fileName,frame)

        else:

            break

    return fps

def img2video(outvideoroot, fps):

    img = cv2.imread('./imgs/image1.jpg')

    imgInfo = img.shape

    size = (imgInfo[1],imgInfo[0])

    print(size)

    fourcc = cv2.VideoWriter_fourcc(*'XVID')

    videowriter = cv2.VideoWriter(outvideoroot,fourcc,fps,size)

    list_imgs = os.listdir('./imgs')

    for i in range(1, len(list_imgs)):

        fileName = './imgs/image'+str(i)+'.jpg'

        img = cv2.imread(fileName)

        videowriter.write(img)

def get_xoy():

    imgsroot = './imgs'

    list_ = os.listdir(imgsroot)

    for n in list_:

        path_ = os.path.join(imgsroot, n)

        img = cv2.imread(path_)

        bbox = cv2.selectROI('selectroi', img)

        break

    return bbox

def fitsign():

    box = get_xoy()

    xo, yo, w, h = box[0], box[1],box[2],box[3],

    print(box)

    dstroot = './imgs'

    dstlist = os.listdir(dstroot)

    kernel_size = (41, 41)

    sigma = 50

    for n in dstlist:

        path_ = os.path.join(dstroot, n)

        img = cv2.imread(path_)

        crop = img[yo:yo+h, xo:xo+w, :]

        crop = cv2.GaussianBlur(crop, kernel_size, sigma)

       # crop = cv2.blur(crop, (41,41))

        img[yo:yo+h, xo:xo+w, :] = crop

        cv2.imwrite(path_, img)

if __name__ == "__main__":

    fps = video2img("222056.mov")

    fitsign()

    img2video('a.avi', fps)