From 63f5a744f166208889f9b889d14ea48d5ce0cba9 Mon Sep 17 00:00:00 2001 From: nunenuh Date: Sun, 14 Aug 2022 05:39:35 +0800 Subject: [PATCH] fix: fixing error when converting dataset 300W in class Landmarks300WConverter #60 --- torchlm/data/_converters.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torchlm/data/_converters.py b/torchlm/data/_converters.py index fe86eb6..c1868c8 100644 --- 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)