Skip to content
Snippets Groups Projects
Commit 3819502d authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Cleaning up

parent e2440230
No related branches found
No related tags found
1 merge request!112Feature extractors
Pipeline #51512 passed
import bob.bio.base
from bob.bio.face.preprocessor import FaceCrop
from bob.bio.face.annotator import BobIpTinyface
from bob.bio.face.extractor import MxNetModel
from bob.bio.base.algorithm import Distance
from bob.bio.base.pipelines.vanilla_biometrics.legacy import BioAlgorithmLegacy
import scipy.spatial
from bob.bio.base.pipelines.vanilla_biometrics import Distance
from sklearn.pipeline import make_pipeline
from bob.pipelines import wrap
from bob.bio.base.pipelines.vanilla_biometrics import VanillaBiometricsPipeline
annotator_transformer = BobIpTinyface()
preprocessor_transformer = FaceCrop(
cropped_image_size=(112, 112),
cropped_positions={"leye": (49, 72), "reye": (49, 38)},
color_channel="rgb",
annotator=annotator_transformer,
)
extractor_transformer = MxNetModel()
algorithm = Distance(
distance_function=scipy.spatial.distance.cosine, is_distance_function=True
)
transformer = make_pipeline(
wrap(["sample"], preprocessor_transformer), wrap(["sample"], extractor_transformer)
)
pipeline = VanillaBiometricsPipeline(transformer, algorithm)
transformer = pipeline.transformer
...@@ -137,7 +137,6 @@ setup( ...@@ -137,7 +137,6 @@ setup(
"resnet50-msceleb-arcface-20210521 = bob.bio.face.config.baseline.resnet50_msceleb_arcface_20210521:pipeline", "resnet50-msceleb-arcface-20210521 = bob.bio.face.config.baseline.resnet50_msceleb_arcface_20210521:pipeline",
"resnet50-vgg2-arcface-2021 = bob.bio.face.config.baseline.resnet50_vgg2_arcface_2021:pipeline", "resnet50-vgg2-arcface-2021 = bob.bio.face.config.baseline.resnet50_vgg2_arcface_2021:pipeline",
"mobilenetv2-msceleb-arcface-2021 = bob.bio.face.config.baseline.mobilenetv2_msceleb_arcface_2021:pipeline", "mobilenetv2-msceleb-arcface-2021 = bob.bio.face.config.baseline.mobilenetv2_msceleb_arcface_2021:pipeline",
"mxnet-tinyface = bob.bio.face.config.baseline.mxnet_tinyface:pipeline",
"afffe = bob.bio.face.config.baseline.afffe:pipeline", "afffe = bob.bio.face.config.baseline.afffe:pipeline",
"vgg16-oxford = bob.bio.face.config.baseline.vgg16_oxford:pipeline", "vgg16-oxford = bob.bio.face.config.baseline.vgg16_oxford:pipeline",
"iresnet34 = bob.bio.face.config.baseline.iresnet34:pipeline", "iresnet34 = bob.bio.face.config.baseline.iresnet34:pipeline",
...@@ -154,8 +153,6 @@ setup( ...@@ -154,8 +153,6 @@ setup(
"arcface-insightface = bob.bio.face.config.baseline.arcface_insightface", "arcface-insightface = bob.bio.face.config.baseline.arcface_insightface",
"lgbphs = bob.bio.face.config.baseline.lgbphs", "lgbphs = bob.bio.face.config.baseline.lgbphs",
"lda = bob.bio.face.config.baseline.lda", "lda = bob.bio.face.config.baseline.lda",
"mxnet-pipe = bob.bio.face.config.baseline.mxnet_pipe",
"mxnet-tinyface = bob.bio.face.config.baseline.mxnet_tinyface",
"afffe = bob.bio.face.config.baseline.afffe", "afffe = bob.bio.face.config.baseline.afffe",
"vgg16-oxford = bob.bio.face.config.baseline.vgg16_oxford", "vgg16-oxford = bob.bio.face.config.baseline.vgg16_oxford",
"arface = bob.bio.face.config.database.arface", "arface = bob.bio.face.config.database.arface",
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment