babysister.frames_reader module

Frames reading routines

exception babysister.frames_reader.FrameReadError

Bases: Exception

Error that will be raised when a frame reader cannot read.

class babysister.frames_reader.ImagesReader(frames_dir, im_ext='jpg')

Bases: object

Reader for sequence of frames.

Parameters:

• frames_dir (str) – path to folder that contains frames.
• im_ext (str) – image extension.

frames_path

full path of images with im_ext extension in frames_dir.

Type:list of str

get_frame_size()

Get size of the first image in frames_path.

read()

Try to read a frame from frames_path.

Returns:frame data in BGR format. Return type:ndarray Raises:FrameReadError – if cannot read.

class babysister.frames_reader.VideoReader(video_path)

Bases: object

Reader for video.

Parameters:video_path (str) – path to video.

cap

VideoCapture of video_path.

Type:cv.VideoCapture

get_frame_size()

Get size of video.

read()

Try to read a frame from cap.

Returns:frame data in BGR format. Return type:ndarray Raises:FrameReadError – if cannot read.