Merge pull request #62 from DefTruth/dev

fix(converters): fixed error when converting dataset 300W (#60)
DefTruth · Aug 14, 2022 · df5a6af · df5a6af
2 parents 71d978a + ce9cb0f
commit df5a6af
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)