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
No related branches found
No related tags found
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.
Finish editing this message first!
Please register or to comment