Skip to content
Snippets Groups Projects
Commit 72035ee5 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

Add the giant video loader

parent f3d9701f
No related branches found
No related tags found
1 merge request!60Change the API of yield_faces
from .load_utils import (
frames, number_of_frames, yield_frames, yield_faces, scale_face, blocks,
bbx_cropper, min_face_size_normalizer)
bbx_cropper, min_face_size_normalizer, the_giant_video_loader)
# gets sphinx autodoc done right - don't remove it
__all__ = [_ for _ in dir() if not _.startswith('_')]
......@@ -5,7 +5,6 @@ from bob.ip.base import scale, block, block_output_shape
from bob.ip.facedetect import bounding_box_from_annotation
from functools import partial
import numpy
import six
def frames(path):
......@@ -191,3 +190,18 @@ def blocks(data, block_size, block_overlap=(0, 0)):
else:
raise ValueError("Unknown data dimension {}".format(data.ndim))
return output
def the_giant_video_loader(paddb, padfile,
region='whole', scaling_factor=None, cropper=None,
normalizer=None):
generator = None
if region == 'whole':
generator = yield_frames(paddb, padfile)
elif region == 'crop':
generator = yield_faces(
paddb, padfile, cropper=cropper, normalizer=normalizer)
if scaling_factor is not None:
generator = (scale(frame, scaling_factor)
for frame in generator)
return generator
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment