Python Video Stabilizer с использованием фильтра низких частот - PullRequest
Новая
       86

Python Video Stabilizer с использованием фильтра низких частот

0 голосов
/ 25 января 2019

У меня есть проект, в котором я должен реализовать стабилизатор видео, используя fft и фильтры (lpf или hpf)

Вот часть кода, которую я хочу изменить: 

import cv2
import numpy as np

# Create a VideoCapture object and read from input file
cap = cv2.VideoCapture('Vibrated2.avi')

# load data
data = np.loadtxt('Vibrated2.txt', delimiter=',')

# Define the codec and create VideoWriter object
frame_width = int(cap.get(3))
frame_height = int(cap.get(4))
out = cv2.VideoWriter('Stabilized.avi',cv2.VideoWriter_fourcc('M','J','P','G'), 25, (frame_width,frame_height))

# Check if camera opened successfully
if (cap.isOpened()== False): 
  print("Error opening video stream or file")

def transform(frame, param):
  ti = cv2.getRotationMatrix2D((0,0), 0, 1)
  ti[0,2] += param[0]
  ti[1,2] += param[1]
  frame = cv2.warpAffine(frame, ti, frame.shape[1:-4:-1])
  return frame

num = 0 
# Read until video is completed
while(cap.isOpened()):
  # Capture frame-by-frame
  ret, frame = cap.read()
  if ret == True:
    # apply transformation
    frame = transform(frame, data[num])
    print (frame, "dn")
    num+=1

    # Display the resulting frame
    cv2.imshow('Frame',frame)

    # Write the frame into the file 'output.avi'
    out.write(frame)

    # Press Q on keyboard to  exit
    if cv2.waitKey(25) & 0xFF == ord('q'):
      break

  # Break the loop
  else: 
    break

# When everything done, release the video capture object
cap.release()

# Closes all the frames
cv2.destroyAllWindows()

Существует видео с именем Vibrated1.avi и текстовый файл, который включает в себя разницу в кадрах с именем Virated1.txt, и выглядит это так:

0.341486, -0.258215

0.121945, 1.27605

-0.0811261, 0.78985

, ...

Я не знаю, как и где мне добавить несколько фильтров в этот код для удаления вибрации видео

Кто-нибудь может мне помочь?

1 Ответ

0 голосов
/ 25 января 2019

Я могу написать короткий псевдокод с частями C ++: 

cv::Mat homoFiltered = cv::Mat::eye(3, 3, CV_32F);
const double alpha = 0.9;
cv::Mat a1 = cv::Mat::eye(3, 3, CV_32F) * alpha;
cv::Mat a2 = cv::Mat::eye(3, 3, CV_32F) * (1. - alpha);


while cap >> frame:
    cv::Mat homo = CalcHomography(frame)
    homoFiltered = a1 * (homoFiltered * homo) + a2 * homo;
    cv::warpPerspective(..., homoFiltered, ...)

альфа - коэффициент сглаживания низких частот в [0; 1]

a1 и a2 - матрицы для применения альфа и (1 - альфа) к матрице преобразования

homoFiltered - отфильтрованная матрица преобразования

гомо - текущая матрица между кадром (t-1) и кадром (t)

...