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

Merge branch 'bio' into 'master'

dont print use logger, try again if video fails to load

See merge request !5
parents caa5e209 0a1efca3
Branches
Tags
1 merge request!5dont print use logger, try again if video fails to load
Pipeline #
......@@ -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():
......
......@@ -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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment