Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

DO NOT MERGE tripod mode #111

Open
wants to merge 2 commits into
base: master
Choose a base branch
from
+13 −2
Open

DO NOT MERGE tripod mode #111

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
933544c
add 1st pass tripod mode
AdamSpannbauer Jun 12, 2020
54a4e8f
Merge branch 'master' into tripod#102
AdamSpannbauer Jun 25, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 13 additions & 2 deletions vidstab/VidStab.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -62,6 +62,11 @@ class VidStab:
:ivar trajectory: a 2d showing the trajectory of the input video
:ivar smoothed_trajectory: a 2d numpy array showing the smoothed trajectory of the input video
:ivar transforms: a 2d numpy array storing the transformations used from frame to frame
:ivar tripod_mode: a bool indicating if tripod_mode is on. tripod_mode will lock the reference frame
for stabilization to be the first frame in video stream. For use w/ locked camera
positions. The assumption is that there will be less movement of camera and that
movement of objects in frame should not be the focus of the stabilization, but rather
adjusting view to the original camera FOV.
"""

def __init__(self, kp_method='GFTT', processing_max_dim=float('inf'), *args, **kwargs):
Expand Down Expand Up @@ -108,6 +113,7 @@ def __init__(self, kp_method='GFTT', processing_max_dim=float('inf'), *args, **k

self.frame_queue = FrameQueue()
self.prev_kps = self.prev_gray = None
self.tripod_mode = False

self.writer = None

Expand All @@ -123,6 +129,7 @@ def __init__(self, kp_method='GFTT', processing_max_dim=float('inf'), *args, **k

self._default_stabilize_frame_output = None


def _resize_frame(self, frame):
if self._processing_resize_kwargs == {}:
if self.processing_max_dim:
Expand Down Expand Up @@ -170,7 +177,9 @@ def _gen_next_raw_transform(self):
transform_i = vidstab_utils.estimate_partial_transform(matched_keypoints)

# update previous frame info for next iteration
self._update_prev_frame(current_frame_gray)
if not self.tripod_mode:
self._update_prev_frame(current_frame_gray)

self._raw_transforms.append(transform_i[:])
self._update_trajectory(transform_i)

Expand Down Expand Up @@ -503,7 +512,8 @@ def stabilize_frame(self, input_frame, smoothing_window=30,

def stabilize(self, input_path, output_path, smoothing_window=30, max_frames=float('inf'),
border_type='black', border_size=0, layer_func=None, playback=False,
use_stored_transforms=False, show_progress=True, output_fourcc='MJPG'):
use_stored_transforms=False, show_progress=True, output_fourcc='MJPG',
tripod_mode=False):
"""Read video, perform stabilization, & write stabilized video to file

:param input_path: Path to input video to stabilize.
Expand Down Expand Up @@ -540,6 +550,7 @@ def stabilize(self, input_path, output_path, smoothing_window=30, max_frames=flo
>>> stabilizer = VidStab(kp_method = 'ORB')
>>> stabilizer.stabilize(input_path='input_video.mov', output_path='stable_video.avi')
"""
self.tripod_mode = tripod_mode
self.writer = None

if border_size == 'auto':
Expand Down