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

Small modifications in BATL HLDI

parent 08484e67
Branches
Tags
2 merge requests!79Batl hldi update,!78Small modifications in BATL HLDI
Pipeline #25977 canceled
...@@ -573,6 +573,14 @@ class BatlPadDatabase(PadDatabase): ...@@ -573,6 +573,14 @@ class BatlPadDatabase(PadDatabase):
if not os.path.isfile(file_path): # no file with annotations 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.stream_type = "color"
f.reference_stream_type = "color" f.reference_stream_type = "color"
f.warp_to_reference = False f.warp_to_reference = False
...@@ -583,6 +591,14 @@ class BatlPadDatabase(PadDatabase): ...@@ -583,6 +591,14 @@ class BatlPadDatabase(PadDatabase):
video = f.load(directory=self.original_directory, video = f.load(directory=self.original_directory,
extension=self.original_extension) 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 = {} annotations = {}
for idx, image in enumerate(video.as_array()): for idx, image in enumerate(video.as_array()):
...@@ -614,7 +630,7 @@ class BatlPadDatabase(PadDatabase): ...@@ -614,7 +630,7 @@ class BatlPadDatabase(PadDatabase):
# If specified append annotations for the roi in the facial region: # If specified append annotations for the roi in the facial region:
if self.append_color_face_roi_annot: 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: with open(file_path, 'r') as json_file: # open the file containing all annotations:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment