From 9ee22572418bba86b78e6059d29beaeb6111f6f2 Mon Sep 17 00:00:00 2001
From: Guillaume HEUSCH <guillaume.heusch@idiap.ch>
Date: Mon, 14 Jan 2019 16:12:00 +0100
Subject: [PATCH] started high-level implementation for CASIA-SURF DB
---
bob/pad/face/database/casiasurf.py | 76 ++++++++++++++++++++++++++++++
1 file changed, 76 insertions(+)
create mode 100644 bob/pad/face/database/casiasurf.py
diff --git a/bob/pad/face/database/casiasurf.py b/bob/pad/face/database/casiasurf.py
new file mode 100644
index 00000000..ac6b7e3a
--- /dev/null
+++ b/bob/pad/face/database/casiasurf.py
@@ -0,0 +1,76 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+
+import os
+import numpy as np
+import bob.io.video
+from bob.bio.video import FrameSelector, FrameContainer
+from bob.pad.face.database import VideoPadFile
+from bob.pad.base.database import PadDatabase
+
+class CasiaSurfPadFile(VideoPadFile):
+ """
+ A high level implementation of the File class for the CASIA-SURF database.
+
+ Note that this does not represent a file per se, but rather a sample
+ that may contain more than one file.
+
+ Attributes
+ ----------
+ f : :py:class:`object`
+ An instance of the Sample class defined in the low level db interface
+ of the CASIA-SURF database, in the bob.db.casiasurf.models.py file.
+
+ """
+
+ def __init__(self, f, stream_type):
+ """ Init
+
+ Parameters
+ ----------
+ f : :py:class:`object`
+ An instance of the Sample class defined in the low level db interface
+ of the CASIA-SURF database, in the bob.db.casiasurf.models.py file.
+ stream_type: str of list of str
+ The streams to be loaded.
+ """
+ self.f = f
+ self.stream_type = stream_type
+ super(CasiaSurfPadFile, self).__init__(
+ client_id=f.id,
+ attack_type=attack_type)
+
+
+ def load(self, directory=None, extension='.jpg', frame_selector=FrameSelector(selection_style='all')):
+ """Overridden version of the load method defined in ``VideoPadFile``.
+
+ Parameters
+ ----------
+ directory : :py:class:`str`
+ String containing the path to the CASIA-SURF database
+ extension : :py:class:`str`
+ Extension of the image files
+ frame_selector : :py:class:`bob.bio.video.FrameSelector`
+ The frame selector to use.
+
+ Returns
+ -------
+ dict:
+ "video" data (i.e. one frame) for multiple streams stored in the dictionary.
+ The structure of the dictionary: ``data={"stream1_name" : FrameContainer1, "stream2_name" : ...}``
+ Names of the streams are defined in ``self.stream_type``.
+ """
+
+ # get the dict of numpy array
+ data = self.f.load(directory, extension, modality=self.modality)
+
+ # convert that to dict of FrameContainer
+ data_to_return = {}
+ for k in data.keys():
+ frame_container = FrameContainer()
+ for idx, item in enumerate(data[k]):
+ frame_container.add(idx, item) # add frame to FrameContainer
+ data_to_return[k] = frame_container
+
+ return data_to_return
--
GitLab