diff --git a/bob/db/replaymobile/models.py b/bob/db/replaymobile/models.py index 86bf4747e08bab34eabc3723d98cdf544fa7a578..c623b815b71679ba80e9cc6d6eac96a829929b44 100644 --- a/bob/db/replaymobile/models.py +++ b/bob/db/replaymobile/models.py @@ -12,6 +12,9 @@ from sqlalchemy.ext.declarative import declarative_base import numpy import bob.io.base import bob.io.video +import bob.core + +logger = bob.core.log.setup('bob.db.replaymobile') Base = declarative_base() @@ -206,7 +209,7 @@ class File(Base): [optional] The extension of the filename - this will control the type of output and the codec for saving the input blob. """ - print('video file extension:', extension) + logger.debug('video file extension: {}'.format(extension)) if extension is None: extension = '.mov' # if self.get_quality() == 'laptop': @@ -223,7 +226,7 @@ class File(Base): vin = numpy.rollaxis(vin, 3, 2) if not self.is_tablet(): - print("flipping mobile video") + logger.debug('flipping mobile video') vin = vin[:, :, ::-1, :] # if self.is_rotated(): diff --git a/bob/db/replaymobile/verificationprotocol.py b/bob/db/replaymobile/verificationprotocol.py index 405c2c851fa6a0bbb5d2cc90f085b10cc9a06336..845b4a0207ad2a378db57eafc172436f754d0f51 100644 --- a/bob/db/replaymobile/verificationprotocol.py +++ b/bob/db/replaymobile/verificationprotocol.py @@ -35,9 +35,13 @@ class File(BaseFile): def load(self, directory=None, extension=None): if extension in (None, '.mov'): - video = self._f.load(directory, extension) - # just return the required frame. - return video[self.framen] + for i in range(100): + try: + video = self._f.load(directory, extension) + # just return the required frame. + return video[self.framen] + except RuntimeError: + pass else: return super(File, self).load(directory, extension)