From 81d2d597b86e0000d38b1d89f0f2fa7f22a38f77 Mon Sep 17 00:00:00 2001 From: Olegs NIKISINS <onikisins@italix03.idiap.ch> Date: Mon, 12 Mar 2018 16:53:28 +0100 Subject: [PATCH] Changed the annotations to work with HLDI load method --- bob/pad/face/database/batl.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/bob/pad/face/database/batl.py b/bob/pad/face/database/batl.py index 6a3fda6a..fe44953d 100644 --- a/bob/pad/face/database/batl.py +++ b/bob/pad/face/database/batl.py @@ -248,14 +248,15 @@ class BatlPadDatabase(PadDatabase): if not os.path.isfile(file_path): # no file with annotations - video = f.f.load(directory=self.original_directory, - extension=self.original_extension, - modality="color", # TODO: this parameter is currently missing in bob.db.batl, add it there - reference_stream_type="color", - warp_to_reference=False, - convert_to_rgb=False, - crop=None, - max_frames=None)['video'] + f.stream_type = "color" + f.reference_stream_type = "color" + f.warp_to_reference = False + f.convert_to_rgb = False + f.crop = None + f.video_data_only = True + + video = f.load(directory=self.original_directory, + extension=self.original_extension) annotations = {} -- GitLab