fix: fixing error when converting dataset 300W in class Landmarks300W…

…Converter DefTruth#60
nunenuh · Aug 13, 2022 · 63f5a74 · 63f5a74
1 parent 71d978a
commit 63f5a74
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torchlm/data/_converters.py b/torchlm/data/_converters.py
@@ -490,8 +490,8 @@ def _process_annotation(
         # crop padded face
         crop = image[int(ymin):int(ymax), int(xmin):int(xmax), :]
         # adjust according to left-top corner
-        landmarks[:, 0] -= float(xmin)
-        landmarks[:, 1] -= float(ymin)
+        landmarks[:, 0] = landmarks[:, 0] - float(xmin)
+        landmarks[:, 1] = landmarks[:, 1] - float(ymin)
 
         if self.target_size is not None and self.resize_op is not None:
             crop, landmarks = self.resize_op(crop, landmarks)