From ffaec1f370a35d37231cffd6819eb889f27f3ff7 Mon Sep 17 00:00:00 2001
From: Olegs NIKISINS <onikisins@italix03.idiap.ch>
Date: Tue, 15 Jan 2019 17:18:47 +0100
Subject: [PATCH] Small modifications in BATL HLDI
---
bob/pad/face/database/batl.py | 18 +++++++++++++++++-
1 file changed, 17 insertions(+), 1 deletion(-)
diff --git a/bob/pad/face/database/batl.py b/bob/pad/face/database/batl.py
index 99a2ec28..f310f62f 100644
--- a/bob/pad/face/database/batl.py
+++ b/bob/pad/face/database/batl.py
@@ -573,6 +573,14 @@ class BatlPadDatabase(PadDatabase):
if not os.path.isfile(file_path): # no file with annotations
+ # original values of the arguments of f:
+ stream_type_original = f.stream_type
+ reference_stream_type_original = f.reference_stream_type
+ warp_to_reference_original = f.warp_to_reference
+ convert_to_rgb_original = f.convert_to_rgb
+ crop_original = f.crop
+ video_data_only_original = f.video_data_only
+
f.stream_type = "color"
f.reference_stream_type = "color"
f.warp_to_reference = False
@@ -583,6 +591,14 @@ class BatlPadDatabase(PadDatabase):
video = f.load(directory=self.original_directory,
extension=self.original_extension)
+ # set arguments of f to the original values:
+ f.stream_type = stream_type_original
+ f.reference_stream_type = reference_stream_type_original
+ f.warp_to_reference = warp_to_reference_original
+ f.convert_to_rgb = convert_to_rgb_original
+ f.crop = crop_original
+ f.video_data_only = video_data_only_original
+
annotations = {}
for idx, image in enumerate(video.as_array()):
@@ -614,7 +630,7 @@ class BatlPadDatabase(PadDatabase):
# If specified append annotations for the roi in the facial region:
if self.append_color_face_roi_annot:
- file_path = pkg_resources.resource_filename( 'bob.pad.face', os.path.join('lists/batl/color_skin_non_skin_annotations/', "annotations_train_set" + ".json") )
+ file_path = pkg_resources.resource_filename( 'bob.pad.face', os.path.join('lists/batl/color_skin_non_skin_annotations/', "annotations_train_dev_set" + ".json") )
with open(file_path, 'r') as json_file: # open the file containing all annotations:
--
GitLab