From d1754b934fd3819cb9340d9a0175a75a9e321996 Mon Sep 17 00:00:00 2001
From: Ivana Chingovska <ivana.chingovska@idiap.ch>
Date: Thu, 13 Sep 2012 16:54:37 +0200
Subject: [PATCH] added methods for querying the face locations bbx
---
xbob/db/replay/query.py | 74 +++++++++++++++++++++++++++++++++++++----
xbob/db/replay/test.py | 4 +--
2 files changed, 70 insertions(+), 8 deletions(-)
diff --git a/xbob/db/replay/query.py b/xbob/db/replay/query.py
index e650ec3..20ac027 100644
--- a/xbob/db/replay/query.py
+++ b/xbob/db/replay/query.py
@@ -156,7 +156,7 @@ class Database(object):
return retval
- def faces(self, filenames, directory=None):
+ def facefiles(self, filenames, directory=None):
"""Queries the files containing the face locations for the frames in the videos specified by the input parameter filenames
Keyword parameters:
@@ -178,11 +178,42 @@ class Database(object):
There is one row for each frame, and not all the frames contain detected faces
"""
- if directory: return [os.path.join(directory, stem + '.faces') for stem in filenames]
- return [stem + '.faces' for stem in filenames]
+ if directory: return [os.path.join(directory, stem + '.face') for stem in filenames]
+ return [stem + '.face' for stem in filenames]
-
- def faces_ids(self, ids, directory=None):
+ def facebbx(self, filenames, directory=None):
+ """Reads the files containing the face locations for the frames in the videos specified by the input parameter filenames
+
+ Keyword parameters:
+
+ filenames
+ The filenames of the videos. This object should be a python iterable (such as a tuple or list).
+
+ Returns:
+ A list of numpy.ndarrays containing information about the locatied faces in the videos. Each element in the list corresponds to one input filename. Each row of the numpy.ndarray corresponds for one frame. The five columns of the numpy.ndarray denote:
+
+ * Frame number
+ * Bounding box top-left X coordinate
+ * Bounding box top-left Y coordinate
+ * Bounding box width
+ * Bounding box height
+
+ Note that not all the frames contain detected faces.
+ """
+ facefiles = self.facefiles(filenames, directory)
+ facesbbx = []
+ for facef in facefiles:
+ lines = open(facef, "r").readlines()
+ bbx = numpy.ndarray((len(lines), 5), dtype='int')
+ lc = 0
+ for l in lines:
+ words = l.split()
+ bbx[lc] = [int(w) for w in words]
+ lc+=1
+ facesbbx.append(bbx)
+ return facesbbx
+
+ def facefiles_ids(self, ids, directory=None):
"""Queries the files containing the face locations for the frames in the videos specified by the input parameter ids
Keyword parameters:
@@ -198,9 +229,40 @@ class Database(object):
"""
if not directory:
directory = ''
- facespaths = self.paths(ids, prefix=directory, suffix='.faces')
+ facespaths = self.paths(ids, prefix=directory, suffix='.face')
return facespaths
+ def facebbx_ids(self, ids, directory=None):
+ """Reads the files containing the face locations for the frames in the videos specified by the input parameter filenames
+
+ Keyword parameters:
+
+ filenames
+ The filenames of the videos. This object should be a python iterable (such as a tuple or list).
+
+ Returns:
+ A list of numpy.ndarrays containing information about the locatied faces in the videos. Each element in the list corresponds to one input filename. Each row of the numpy.ndarray corresponds for one frame. The five columns of the numpy.ndarray denote:
+
+ * Frame number
+ * Bounding box top-left X coordinate
+ * Bounding box top-left Y coordinate
+ * Bounding box width
+ * Bounding box height
+
+ Note that not all the frames contain detected faces.
+ """
+ facefiles = self.facefiles_ids(ids, directory)
+ facesbbx = []
+ for facef in facefiles:
+ lines = open(facef, "r").readlines()
+ bbx = numpy.ndarray((len(lines), 5), dtype='int')
+ lc = 0
+ for l in lines:
+ words = l.split()
+ bbx[lc] = [int(w) for w in words]
+ lc+=1
+ facesbbx.append(bbx)
+ return facesbbx
def protocols(self):
"""Returns the names of all registered protocols"""
diff --git a/xbob/db/replay/test.py b/xbob/db/replay/test.py
index ad13189..8e0e8ee 100644
--- a/xbob/db/replay/test.py
+++ b/xbob/db/replay/test.py
@@ -131,8 +131,8 @@ class ReplayDatabaseTest(unittest.TestCase):
def test13_queryfacefile(self):
db = Database()
- self.assertEqual(db.faces(('foo',), directory = 'dir')[0], 'dir/foo.faces',)
+ self.assertEqual(db.facefiles(('foo',), directory = 'dir')[0], 'dir/foo.face',)
def test14_queryfacefile_key(self):
db = Database()
- self.assertEqual(db.faces_ids(ids=(1,), directory='dir'), db.paths(ids=(1,), prefix='dir', suffix='.faces'))
+ self.assertEqual(db.facefiles_ids(ids=(1,), directory='dir'), db.paths(ids=(1,), prefix='dir', suffix='.face'))
--
GitLab