From 6afa6a9eb552973506e98f3b87ae84bf4cd62d11 Mon Sep 17 00:00:00 2001
From: torzdf <36920800+torzdf@users.noreply.github.com>
Date: Mon, 9 Dec 2019 17:54:35 +0000
Subject: [PATCH] bugfix: plugins.train.trainer._base - Fix landmarks loading

---
 plugins/train/trainer/_base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/plugins/train/trainer/_base.py b/plugins/train/trainer/_base.py
index 981e9fbc54..04ae7be21d 100644
--- 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