Commit 03490c05 authored by Guillaume HEUSCH's avatar Guillaume HEUSCH
Browse files

added fargo database

parent 180bac41
Pipeline #24188 passed with stage
in 18 minutes and 23 seconds
......@@ -20,6 +20,9 @@ from .scface import SCFaceBioDatabase
from .replaymobile import ReplayMobileBioDatabase
from .msu_mfsd_mod import MsuMfsdModBioDatabase
from .fargo import FargoBioDatabase
# gets sphinx autodoc done right - don't remove it
......
#!/usr/bin/env python
# vim: set fileencoding=utf-8 :
"""
FARGO database implementation of bob.bio.base.database.Database interface.
It is an extension of an SQL-based database interface, which directly talks to FARGO database, for
verification experiments (good to use in bob.bio.base framework).
"""
import os
import bob.db.base
from .database import FaceBioFile
from bob.bio.base.database import BioDatabase
class FargoBioDatabase(BioDatabase):
"""
FARGO database implementation of :py:class:`bob.bio.base.database.BioDatabase` interface.
It is an extension of the database interface, which directly talks to ATNT database, for
verification experiments (good to use in bob.bio.base framework).
"""
def __init__(
self,
original_directory=None,
original_extension='.png',
protocol='mc-rgb',
**kwargs
):
# call base class constructors to open a session to the database
super(FargoBioDatabase, self).__init__(
name='fargo',
original_directory=original_directory,
original_extension=original_extension,
protocol=protocol,
**kwargs)
from bob.db.fargo.query import Database as LowLevelDatabase
self._db = LowLevelDatabase(original_directory, original_extension, protocol=protocol)
def objects(self, groups=None, purposes=None, protocol=None, model_ids=None, **kwargs):
retval = self._db.objects(protocol=protocol, groups=groups, purposes=purposes, model_ids=model_ids, **kwargs)
return [FaceBioFile(client_id=f.client_id, path=f.path, file_id=f.id) for f in retval]
def model_ids_with_protocol(self, groups=None, protocol=None, **kwargs):
return self._db.model_ids(groups=groups, protocol=protocol)
def annotations(self, file):
if self.annotation_directory is None:
return None
annotation_file = os.path.join(self.annotation_directory, file.path + self.annotation_extension)
return bob.db.base.read_annotation_file(annotation_file, 'eyecenter')
......@@ -136,6 +136,8 @@ setup(
'replay-img-spoof = bob.bio.face.config.database.replay:replay_spoof',
'replaymobile-img-licit = bob.bio.face.config.database.replaymobile:replaymobile_licit',
'replaymobile-img-spoof = bob.bio.face.config.database.replaymobile:replaymobile_spoof',
'fargo = bob.bio.face.config.database.fargo:database',
],
'bob.bio.annotator': [
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment