diff --git a/bob/bio/face/config/database/rfw.py b/bob/bio/face/config/database/rfw.py new file mode 100644 index 0000000000000000000000000000000000000000..3b350bfe8965a53ac76c6ef455f98536e57075d4 --- /dev/null +++ b/bob/bio/face/config/database/rfw.py @@ -0,0 +1,9 @@ +from bob.bio.face.database import RFWDatabase + +# 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 = "original" + + +database = RFWDatabase(protocol=protocol) diff --git a/setup.py b/setup.py index 9e789e233ea96d06dc92981cdad210ff0f1f1ffe..4554ca22617150b6f3c78599eb556d02ace58562 100644 --- a/setup.py +++ b/setup.py @@ -110,6 +110,7 @@ setup( "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", + "rfw = bob.bio.face.config.database.rfw:database", ], "bob.bio.annotator": [ "facedetect = bob.bio.face.config.annotator.facedetect:annotator", @@ -165,6 +166,7 @@ setup( "replaymobile-img = bob.bio.face.config.database.replaymobile", "fargo = bob.bio.face.config.database.fargo", "meds = bob.bio.face.config.database.meds", + "rfw = bob.bio.face.config.database.rfw", "casia-africa = bob.bio.face.config.database.casia_africa", "morph = bob.bio.face.config.database.morph", "resnet50-msceleb-arcface-2021 = bob.bio.face.config.baseline.resnet50_msceleb_arcface_2021",