Skip to content
Snippets Groups Projects
Commit 4791a034 authored by Olegs NIKISINS's avatar Olegs NIKISINS
Browse files

Some fixes to match the load method of bob.db.batl

parent d718f12f
No related branches found
No related tags found
1 merge request!59Added HLDI for the BATL database, added optional data normalization in FaceCropAlign
Pipeline #
......@@ -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 = {}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment