Upload directory

aligners/__init__.py

@@ -1,15 +1,16 @@
 from .base import BaseAligner
-from .none import NoneAligner
-from .differentiable_face_aligner import DifferentiableFaceAligner
-from .retinaface_aligner import RetinaFaceAligner
+
 
 def get_aligner(aligner_cfg):
 
     if aligner_cfg.name == 'none':
+        from .none import NoneAligner
         aligner = NoneAligner.from_config(aligner_cfg)
     elif aligner_cfg.name == 'retinaface_aligner':
+        from .retinaface_aligner import RetinaFaceAligner
         aligner = RetinaFaceAligner.from_config(aligner_cfg)
     elif aligner_cfg.name == 'differentiable_face_aligner':
+        from .differentiable_face_aligner import DifferentiableFaceAligner
         aligner = DifferentiableFaceAligner.from_config(aligner_cfg)
     else:
         raise ValueError(f"Unknown classifier: {aligner_cfg.name}")