Commit 627eab0e authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

Move the imports to the init method to make sure nothing breaks

parent dd061d50
from . import Base, bounding_box_to_annotations from . import Base, bounding_box_to_annotations
from bob.ip.facedetect import bounding_box_from_annotation from bob.ip.facedetect import bounding_box_from_annotation
from bob.ip.dlib import DlibLandmarkExtraction
class BobIpDlib(Base): class BobIpDlib(Base):
...@@ -8,6 +7,7 @@ class BobIpDlib(Base): ...@@ -8,6 +7,7 @@ class BobIpDlib(Base):
def __init__(self, **kwargs): def __init__(self, **kwargs):
super(BobIpDlib, self).__init__(**kwargs) super(BobIpDlib, self).__init__(**kwargs)
from bob.ip.dlib import DlibLandmarkExtraction
self.detector = DlibLandmarkExtraction(bob_landmark_format=True) self.detector = DlibLandmarkExtraction(bob_landmark_format=True)
def annotate(self, image, **kwargs): def annotate(self, image, **kwargs):
......
from . import Base, bounding_box_to_annotations from . import Base, bounding_box_to_annotations
from bob.ip.mtcnn import FaceDetector
class BobIpMTCNN(Base): class BobIpMTCNN(Base):
...@@ -7,6 +6,7 @@ class BobIpMTCNN(Base): ...@@ -7,6 +6,7 @@ class BobIpMTCNN(Base):
def __init__(self, **kwargs): def __init__(self, **kwargs):
super(BobIpMTCNN, self).__init__(**kwargs) super(BobIpMTCNN, self).__init__(**kwargs)
from bob.ip.mtcnn import FaceDetector
self.detector = FaceDetector() self.detector = FaceDetector()
def annotate(self, image, **kwargs): def annotate(self, image, **kwargs):
......
Supports Markdown
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