Commit 34e6ee96 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Created interface for CASIA-Africa database

Created interface for CASIA-Africa database
parent 4dc57ca0
from bob.bio.face.database import CasiaAfricaDatabase
# In case protocol is comming from chain loading
# https://www.idiap.ch/software/bob/docs/bob/bob.extension/stable/py_api.html#bob.extension.config.load
if "protocol" not in locals():
protocol = "ID-V-All-Ep1"
database = CasiaAfricaDatabase(protocol=protocol)
......@@ -14,6 +14,7 @@ from .replaymobile import ReplayMobileBioDatabase
from .fargo import FargoBioDatabase
from .meds import MEDSDatabase
from .morph import MorphDatabase
from .casia_africa import CasiaAfricaDatabase
# gets sphinx autodoc done right - don't remove it
......@@ -47,5 +48,6 @@ __appropriate__(
FargoBioDatabase,
MEDSDatabase,
MorphDatabase,
CasiaAfricaDatabase,
)
__all__ = [_ for _ in dir() if not _.startswith("_")]
#!/usr/bin/env python
# vim: set fileencoding=utf-8 :
# Tiago de Freitas Pereira <tiago.pereira@idiap.ch>
"""
CASIA-Face-Africa: database implementation
"""
from bob.bio.base.database import CSVDataset
from bob.bio.base.database import CSVToSampleLoaderBiometrics
from bob.bio.face.database.sample_loaders import EyesAnnotations
from bob.extension import rc
from bob.extension.download import get_file
import bob.io.base
from sklearn.pipeline import make_pipeline
class CasiaAfricaDatabase(CSVDataset):
"""
The Casia-Face-Africa dataset is composed of 1133 identities from different ethical groups in Nigeria.
The capturing locations are:
- Dabai city in Katsina state
- Hotoro in Kano state
- Birget in Kano state
- Gandun Albasa in Kano state
- Sabon Gari inKano state
- Kano State School of Technology
These locations were strategically selected as they are known to have diverse population of local ethnicities.
.. warning::
Only 17 subjects had their images capture in two sessions.
Images were captured during daytime and night using three different cameras:
- C1: Visual Light Camera
- C2: Visual Light Camera
- C3: NIR camera
This dataset interface implemented the three verificatio protocols: "ID-V-All-Ep1", "ID-V-All-Ep2", and "ID-V-All-Ep3"
and they are organized as the following:
+------------------------------------------------------------------------------------+
| Dev. Set |
+------------------+----------------------------+------------+----------+------------+
| protocol name | Cameras (gallery/probe) | Identities | Gallery | Probes |
+==================+============================+============+==========+============+
| ID-V-All-Ep1 | C1/C2 | 1133 | 2455 | 2426 |
+------------------+----------------------------+------------+----------+------------+
| ID-V-All-Ep2 | C1/C3 | 1133 | 2455 | 1171 |
+------------------+----------------------------+------------+----------+------------+
| ID-V-All-Ep3 | C2/C3 | 1133 | 2466 | 1193 |
+------------------+----------------------------+------------+----------+------------+
.. warning::
Use the command below to set the path of the real data::
$ bob config set bob.db.casia-africa.directory [PATH-TO-MEDS-DATA]
.. code-block:: latex
@article{jawad2020,
author = {Jawad, Muhammad and Yunlong, Wang andCaiyong, Wang and Kunbo, Zhang and Zhenan, Sun},
title = {CASIA-Face-Africa: A Large-scale African Face Image Database},
journal = {IEEE Transactions on Information Forensics and Security},
pages = {},
ISSN = {},
year = {},
type = {Journal Article}
}
Example
-------
Fetching biometric references::
>>> from bob.bio.face.database import CasiaAfricaDatabase
>>> database = CasiaAfricaDatabase(protocol="ID-V-All-Ep1")
>>> database.references()
Fetching probes::
>>> from bob.bio.face.database import CasiaAfricaDatabase
>>> database = CasiaAfricaDatabase(protocol="ID-V-All-Ep1")
>>> database.probes()
Parameters
----------
protocol: str
One of the database protocols. Options are "ID-V-All-Ep1", "ID-V-All-Ep2" and "ID-V-All-Ep3"
"""
def __init__(self, protocol):
# Downloading model if not exists
urls = CasiaAfricaDatabase.urls()
filename = get_file(
"casia_face_africa.tar.gz",
urls,
file_hash="324bd69b581477d30606417be8e30d2a",
)
self.annotation_type = "eyes-center"
self.fixed_positions = None
directory = (
rc["bob.db.casia-africa.directory"]
if rc["bob.db.casia-africa.directory "]
else ""
)
super().__init__(
filename,
protocol,
csv_to_sample_loader=make_pipeline(
CSVToSampleLoaderBiometrics(
data_loader=bob.io.base.load,
dataset_original_directory=directory,
extension=".jpg",
reference_id_equal_subject_id=False,
),
EyesAnnotations(),
),
)
@staticmethod
def protocols():
# TODO: Until we have (if we have) a function that dumps the protocols, let's use this one.
return [
"ID-V-All-Ep1",
"ID-V-All-Ep2",
"ID-V-All-Ep3",
]
@staticmethod
def urls():
return [
"https://www.idiap.ch/software/bob/databases/latest/casia_face_africa.tar.gz",
"http://www.idiap.ch/software/bob/databases/latest/casia_face_africa.tar.gz",
]
......@@ -23,9 +23,7 @@ import pytest
import os
import bob.bio.base
from bob.bio.base.test.utils import db_available
from bob.bio.base.test.test_database_implementations import (
check_database,
)
from bob.bio.base.test.test_database_implementations import check_database
import bob.core
from bob.extension.download import get_file
......@@ -72,9 +70,9 @@ def test_arface():
try:
check_database(database, groups=("dev", "eval"))
except IOError as e:
pytest.skip("The database could not queried; probably the db.sql3 file is missing. Here is the error: '%s'"
pytest.skip(
"The database could not queried; probably the db.sql3 file is missing. Here is the error: '%s'"
% e
)
try:
_check_annotations(database)
......@@ -396,3 +394,13 @@ def test_morph():
assert len(database.references(group="eval")) == 6742
assert len(database.probes(group="eval")) == 6553
def test_casia_africa():
from bob.bio.face.database import CasiaAfricaDatabase
database = CasiaAfricaDatabase("ID-V-All-Ep1")
assert len(database.references()) == 2455
assert len(database.probes()) == 2426
......@@ -13,6 +13,7 @@ Databases
.. autosummary::
bob.bio.face.database.ARFaceBioDatabase
bob.bio.face.database.AtntBioDatabase
bob.bio.face.database.CasiaAfricaDatabase
bob.bio.face.database.MobioDatabase
bob.bio.face.database.ReplayBioDatabase
bob.bio.face.database.ReplayMobileBioDatabase
......
......@@ -112,6 +112,7 @@ setup(
"fargo = bob.bio.face.config.database.fargo:database",
"meds = bob.bio.face.config.database.meds:database",
"morph = bob.bio.face.config.database.morph:database",
"casia-africa = bob.bio.face.config.database.casia_africa:database",
],
"bob.bio.annotator": [
"facedetect = bob.bio.face.config.annotator.facedetect:annotator",
......@@ -177,6 +178,7 @@ setup(
"fargo = bob.bio.face.config.database.fargo",
"meds = bob.bio.face.config.database.meds",
"morph = bob.bio.face.config.database.morph",
"casia-africa = bob.bio.face.config.database.casia_africa",
"resnet50-msceleb-arcface-2021 = bob.bio.face.config.baseline.resnet50_msceleb_arcface_2021",
"resnet50-vgg2-arcface-2021 = bob.bio.face.config.baseline.resnet50_vgg2_arcface_2021",
"mobilenetv2-msceleb-arcface-2021 = bob.bio.face.config.baseline.mobilenetv2_msceleb_arcface_2021",
......
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