Commit 0a1efca3 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

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

parent caa5e209
Pipeline #5539 passed with stages
in 4 minutes and 8 seconds
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment