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

Allowing to set prob_thresh

parent 0412ad80
No related branches found
No related tags found
1 merge request!112Feature extractors
Pipeline #51440 passed
...@@ -5,9 +5,15 @@ from . import Base ...@@ -5,9 +5,15 @@ from . import Base
class BobIpTinyface(Base): class BobIpTinyface(Base):
"""Annotator using tinyface in bob.ip.facedetect""" """Annotator using tinyface in bob.ip.facedetect"""
def __init__(self, **kwargs): def __init__(self, prob_thresh=0.5, **kwargs):
super(BobIpTinyface, self).__init__(**kwargs) super(BobIpTinyface, self).__init__(**kwargs)
self.detector = bob.ip.facedetect.tinyface.TinyFacesDetector(prob_thresh=0.5) self.detector = bob.ip.facedetect.tinyface.TinyFacesDetector(
prob_thresh=prob_thresh
)
@property
def prob_thresh(self):
return self.detector.prob_thresh
def annotate(self, image, **kwargs): def annotate(self, image, **kwargs):
"""Annotates an image using tinyface """Annotates an image using tinyface
......
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