基于python 的电子防抖01 ——使用opencv和vidstab

python vidstab库

在csdn上搜索python稳像和EIS或者视频防抖
只有c++ 的资料
很少公开的代码
开启防抖前

开启防抖后

运行此代码前需要安装visdstab库

pip install vidstab -i https://pypi.tuna.tsinghua.edu.cn/simple

我基于官方文档的例子针对自己的情况修改了部分代码并且增加了去除黑边的功能
目前不是很完善
以后又改进会放出链接

默认3秒后开始裁切，需要可以修改
如果时间太短会造成无数据输入报错
此代码是出于初步状态
默认有20fps的延迟，smoothing_windows可以修改
默认膨胀内核大小3

python代码：

import cv2
from vidstab import VidStab, layer_blend
import numpy as np
import time

stabilizer = VidStab()
cap = cv2.VideoCapture(0, cv2.CAP_DSHOW)
kernel = np.ones((3, 3), np.uint8)
t1 = time.time()
#cap.set(3, 1280)  # width=1920
#cap.set(4, 720)  # height=1080

while True:
    x1_, y1_, x2_, y2_ = [], [], [], []
    ret, frame = cap.read()
    if not ret:
        break
    #frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    stabilized_frame = stabilizer.stabilize_frame(input_frame=frame,
                                                  layer_func=layer_blend,
                                                  border_size=100,
                                                  border_type='black',
                                                  smoothing_window=20)
    if stabilized_frame is None:
        break
    t2 = time.time()
    if t2 - t1 > 3:
        stabilized_copy = stabilized_frame.copy()
        #stabilized_copy = cv2.resize(stabilized_copy, (640, 320) interpolation=cv2.INTER_AREA)
        stabilized_copy = cv2.cvtColor(stabilized_copy, cv2.COLOR_BGR2GRAY)
        _, stabilized_copy = cv2.threshold(stabilized_copy, 29, 255, cv2.THRESH_BINARY)
        stabilized_copy = cv2.morphologyEx(stabilized_copy, cv2.MORPH_CLOSE, kernel)
        contours, hierarchy = cv2.findContours(stabilized_copy, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
        if len(contours) > 0:
            for i in range(len(contours)):
                x, y, w, h = cv2.boundingRect(stabilized_copy)
                x1_.append(x)
                y1_.append(y)
                x2_.append(x + w)
                y2_.append(y + h)
            x1 = min(x1_)
            y1 = min(y1_)
            x2 = max(x2_)
            y2 = max(y2_)
        #cv2.rectangle(stabilized_frame, (x1, y1), (x2, y2), (0, 255, 0), 3)
        stabilized_frame = stabilized_frame[y1 + 40:y2 - 40, x1 + 40:x2 - 40, :]
    cv2.imshow('Stabilized Frame', stabilized_frame)
    key = cv2.waitKey(1)
    if key == 27:
        break
cap.release()
cv2.destroyAllWindows()

无注释
评论区交流
特别需要注释我会留意