Merge pull request #61 from nunenuh/main

Fixing Issue #60 Error on Converting 300W dataset in class Landmarks300WConverter with message UFuncTypeError: Cannot cast ufunc 'subtract' output...
DefTruth · Aug 14, 2022 · ce9cb0f · ce9cb0f
2 parents cfc6d87 + 63f5a74
commit ce9cb0f
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)