From 4791a0346cb1eabdfd08822eb930d2f064a9e6f2 Mon Sep 17 00:00:00 2001 From: Olegs NIKISINS <onikisins@italix03.idiap.ch> Date: Fri, 9 Mar 2018 16:30:57 +0100 Subject: [PATCH] Some fixes to match the load method of bob.db.batl --- bob/pad/face/database/batl.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/bob/pad/face/database/batl.py b/bob/pad/face/database/batl.py index caf98574..ff66c799 100644 --- a/bob/pad/face/database/batl.py +++ b/bob/pad/face/database/batl.py @@ -62,7 +62,7 @@ class BatlPadFile(PadFile): data = f.load(self, directory=directory, extension=extension, - stream_type=self.stream_type, # TODO: this parameter is currently missing in bob.db.batl, add it there + modality=self.stream_type, # TODO: this parameter is currently missing in bob.db.batl, add it there reference_stream_type=self.reference_stream_type, data_format_config=self.data_format_config, warp_to_reference=self.warp_to_reference, @@ -238,15 +238,15 @@ class BatlPadDatabase(PadDatabase): if not os.path.isfile(file_path): # no file with annotations - video = f.load(self, directory=self.original_directory, - extension=self.original_extension, - stream_type="color", # TODO: this parameter is currently missing in bob.db.batl, add it there - reference_stream_type="color", - data_format_config=load_data_config(pkg_resources.resource_filename('batl.utils', 'config/idiap_hdf5_data_config.json')), - warp_to_reference=False, - convert_to_rgb=False, - crop=None, - max_frames=None) + video = f.f.load(self, 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", + data_format_config=load_data_config(pkg_resources.resource_filename('batl.utils', 'config/idiap_hdf5_data_config.json')), + warp_to_reference=False, + convert_to_rgb=False, + crop=None, + max_frames=None) annotations = {} -- GitLab