Commit fd317b06 authored by Amir Mohammadi's avatar Amir Mohammadi

Implements the annotations

parent 56821692
Pipeline #6414 failed with stages
in 16 minutes and 7 seconds
...@@ -19,7 +19,7 @@ class File(BioFile): ...@@ -19,7 +19,7 @@ class File(BioFile):
def __init__(self, f): def __init__(self, f):
super(File, self).__init__(client_id=f.client_id, path=f.path, super(File, self).__init__(client_id=f.client_id, path=f.path,
file_id=f.id) file_id=f.id)
self.__f = f self.__f = f
...@@ -32,18 +32,19 @@ class Database(BioDatabase): ...@@ -32,18 +32,19 @@ class Database(BioDatabase):
super(Database, self).__init__(name='utfvp', **kwargs) super(Database, self).__init__(name='utfvp', **kwargs)
from bob.db.utfvp.query import Database as LowLevelDatabase from bob.db.utfvp.query import Database as LowLevelDatabase
self.__db = LowLevelDatabase() self._db = LowLevelDatabase()
def model_ids_with_protocol(self, groups=None, protocol=None, **kwargs): def model_ids_with_protocol(self, groups=None, protocol=None, **kwargs):
protocol = protocol if protocol is not None else self.protocol protocol = protocol if protocol is not None else self.protocol
return self.__db.model_ids(groups=groups, protocol=protocol) return self._db.model_ids(groups=groups, protocol=protocol)
def objects(self, groups=None, protocol=None, purposes=None, def objects(self, groups=None, protocol=None, purposes=None,
model_ids=None, **kwargs): model_ids=None, **kwargs):
retval = self.__db.objects(groups=groups, protocol=protocol, retval = self._db.objects(groups=groups, protocol=protocol,
purposes=purposes, model_ids=model_ids, **kwargs) purposes=purposes, model_ids=model_ids, **kwargs)
return [File(f) for f in retval] return [File(f) for f in retval]
def annotations(self, file):
return None
...@@ -7,81 +7,77 @@ from bob.bio.base.database import BioFile, BioDatabase ...@@ -7,81 +7,77 @@ from bob.bio.base.database import BioFile, BioDatabase
class File(BioFile): class File(BioFile):
""" """
Implements extra properties of vein files for the Vera Fingervein database Implements extra properties of vein files for the Vera Fingervein database
Parameters: Parameters:
f (object): Low-level file (or sample) object that is kept inside f (object): Low-level file (or sample) object that is kept inside
""" """
def __init__(self, f): def __init__(self, f):
super(File, self).__init__(client_id=f.unique_finger_name, path=f.path, super(File, self).__init__(client_id=f.unique_finger_name, path=f.path,
file_id=f.id) file_id=f.id)
self.__f = f self.__f = f
def mask(self):
"""Returns the binary mask from the ROI annotations available"""
def mask(self): from ..preprocessor.utils import poly_to_mask
"""Returns the binary mask from the ROI annotations available"""
from ..preprocessor.utils import poly_to_mask # The size of images in this database is (250, 665) pixels (h, w)
return poly_to_mask((250, 665), self.__f.roi())
# The size of images in this database is (250, 665) pixels (h, w) def load(self, *args, **kwargs):
return poly_to_mask((250, 665), self.__f.roi()) """(Overrides base method) Loads both image and mask"""
image = super(File, self).load(*args, **kwargs)
def load(self, *args, **kwargs): return image, self.mask()
"""(Overrides base method) Loads both image and mask"""
image = super(File, self).load(*args, **kwargs)
return image, self.mask()
class Database(BioDatabase): class Database(BioDatabase):
""" """
Implements verification API for querying Vera Fingervein database. Implements verification API for querying Vera Fingervein database.
""" """
def __init__(self, **kwargs):
super(Database, self).__init__(name='verafinger', **kwargs)
from bob.db.verafinger.query import Database as LowLevelDatabase
self.__db = LowLevelDatabase()
self.low_level_group_names = ('train', 'dev')
self.high_level_group_names = ('world', 'dev')
def __init__(self, **kwargs):
def groups(self): super(Database, self).__init__(name='verafinger', **kwargs)
from bob.db.verafinger.query import Database as LowLevelDatabase
self._db = LowLevelDatabase()
return self.convert_names_to_highlevel(self.__db.groups(), self.low_level_group_names = ('train', 'dev')
self.low_level_group_names, self.high_level_group_names) self.high_level_group_names = ('world', 'dev')
def groups(self):
def client_id_from_model_id(self, model_id, group='dev'): return self.convert_names_to_highlevel(self._db.groups(),
"""Required as ``model_id != client_id`` on this database""" self.low_level_group_names, self.high_level_group_names)
return self.__db.finger_name_from_model_id(model_id) def client_id_from_model_id(self, model_id, group='dev'):
"""Required as ``model_id != client_id`` on this database"""
return self._db.finger_name_from_model_id(model_id)
def model_ids_with_protocol(self, groups=None, protocol=None, **kwargs): def model_ids_with_protocol(self, groups=None, protocol=None, **kwargs):
groups = self.convert_names_to_lowlevel(groups, groups = self.convert_names_to_lowlevel(groups,
self.low_level_group_names, self.high_level_group_names) self.low_level_group_names, self.high_level_group_names)
return self.__db.model_ids(groups=groups, protocol=protocol) return self._db.model_ids(groups=groups, protocol=protocol)
def objects(self, groups=None, protocol=None, purposes=None,
model_ids=None, **kwargs):
def objects(self, groups=None, protocol=None, purposes=None, groups = self.convert_names_to_lowlevel(groups,
model_ids=None, **kwargs): self.low_level_group_names, self.high_level_group_names)
retval = self._db.objects(groups=groups, protocol=protocol,
purposes=purposes, model_ids=model_ids, **kwargs)
groups = self.convert_names_to_lowlevel(groups, return [File(f) for f in retval]
self.low_level_group_names, self.high_level_group_names)
retval = self.__db.objects(groups=groups, protocol=protocol,
purposes=purposes, model_ids=model_ids, **kwargs)
return [File(f) for f in retval] def annotations(self, file):
return None
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