diff --git a/bob/pad/face/database/replay.py b/bob/pad/face/database/replay.py index b5ef50618abe7ecbe956ec9ee4253d7645536ad3..e144511e42f0586f568ee3b3ea192929f27012d5 100644 --- a/bob/pad/face/database/replay.py +++ b/bob/pad/face/database/replay.py @@ -1,12 +1,11 @@ #!/usr/bin/env python2 # -*- coding: utf-8 -*- - # Used in ReplayMobilePadFile class -from bob.bio.video import FrameSelector, FrameContainer from bob.pad.base.database import PadDatabase from bob.pad.face.database import VideoPadFile # Used in ReplayPadFile class from bob.pad.face.utils import frames, number_of_frames +from bob.extension import rc class ReplayPadFile(VideoPadFile): @@ -58,8 +57,8 @@ class ReplayPadDatabase(PadDatabase): self, # grandtest is the default protocol for this database protocol='grandtest', - original_directory=None, - original_extension=None, + original_directory=rc['bob.db.replay.directory'], + original_extension='.mov', **kwargs): """ Parameters diff --git a/bob/pad/face/database/replay_mobile.py b/bob/pad/face/database/replay_mobile.py index e32b975a9a3cfe93315e1ae80e771696f7e820f9..ce56b280b5c89789f210c1ddb6ca0932eb220445 100644 --- a/bob/pad/face/database/replay_mobile.py +++ b/bob/pad/face/database/replay_mobile.py @@ -7,6 +7,7 @@ from bob.pad.base.database import PadDatabase from bob.pad.face.database import VideoPadFile from bob.pad.face.utils import frames, number_of_frames import numpy +from bob.extension import rc # documentation imports import bob.bio.video @@ -89,8 +90,8 @@ class ReplayMobilePadDatabase(PadDatabase): self, # grandtest is the default protocol for this database protocol='grandtest', - original_directory=None, - original_extension=None, + original_directory=rc['bob.db.replaymobile.directory'], + original_extension='.mov', **kwargs): """ Parameters @@ -251,7 +252,7 @@ class ReplayMobilePadDatabase(PadDatabase): vfilename = padfile.make_path( directory=self.original_directory, extension=self.original_extension) - is_not_tablet = not padfile._f.is_tablet() + is_not_tablet = not padfile.f.is_tablet() for frame in frames(vfilename): frame = numpy.rollaxis(frame, 2, 1) if is_not_tablet: