From d718f12fc473dacbcf7f7b070ccc87c57b25d8a4 Mon Sep 17 00:00:00 2001
From: Olegs NIKISINS <onikisins@italix03.idiap.ch>
Date: Fri, 9 Mar 2018 16:09:34 +0100
Subject: [PATCH] Added temp annotations save option to Batl HLDI
---
bob/pad/face/database/batl.py | 46 +++++++++++++++++++++--------------
1 file changed, 28 insertions(+), 18 deletions(-)
diff --git a/bob/pad/face/database/batl.py b/bob/pad/face/database/batl.py
index fe7252ed..caf98574 100644
--- a/bob/pad/face/database/batl.py
+++ b/bob/pad/face/database/batl.py
@@ -12,6 +12,8 @@ from batl.utils.data import load_data_config
from bob.pad.face.preprocessor.FaceCropAlign import detect_face_landmarks_in_image
+import json
+
class BatlPadFile(PadFile):
"""
A high level implementation of the File class for the BATL
@@ -86,7 +88,7 @@ class BatlPadDatabase(PadDatabase):
protocol='grandtest',
original_directory=rc['bob.db.batl.directory'],
original_extension='.h5',
- annotations_temp_dir=None,
+ annotations_temp_dir="",
landmark_detect_method="mtcnn",
**kwargs):
"""
@@ -231,36 +233,44 @@ class BatlPadDatabase(PadDatabase):
return files
def annotations(self, f):
- pass
-# if annotations_temp_dir is None:
+ file_path = os.path.join(self.annotations_temp_dir, f.f.path + ".json")
- 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)
+ if not os.path.isfile(file_path): # no file with annotations
- 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)
- for idx, image in enumerate(video):
+ annotations = {}
- frame_annotations = detect_face_landmarks_in_image(image, method = self.landmark_detect_method)
+ for idx, image in enumerate(video):
- if frame_annotations:
+ frame_annotations = detect_face_landmarks_in_image(image, method = self.landmark_detect_method)
- annotations[str(idx)] = frame_annotations
+ if frame_annotations:
- return annotations
+ annotations[str(idx)] = frame_annotations
+
+ if self.annotations_temp_dir: # if directory is not an empty string
+ with open(file_path, 'w') as json_file:
+ json_file.write(json.dumps(annotations))
+ else: # if file with annotations exists load them from file
+ with open(file_path, 'r') as json_file:
+ annotations = json.load(json_file)
+
+ return annotations
--
GitLab