bugfix: plugins.train.trainer._base - Fix landmarks loading

o2team · Dec 9, 2019 · 6afa6a9 · 6afa6a9
1 parent 4e654eb
commit 6afa6a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/plugins/train/trainer/_base.py b/plugins/train/trainer/_base.py
@@ -1082,7 +1082,7 @@ def _transform_landmarks(self, side, detected_faces):
             The face filenames as keys with the aligned landmarks as value.
         """
         landmarks = dict()
-        for face in detected_faces:
+        for face in detected_faces.values():
             face.load_aligned(None, size=self._training_opts["training_size"])
             for filename in self._hash_to_filenames(side, face.hash):
                 landmarks[filename] = face.aligned_landmarks