update

2023-12-17 06:53:25 +08:00
parent 3f70c7eb6b
commit 2763361b55
1 changed files with 1 additions and 1 deletions
@@ -456,7 +456,7 @@ class PreProcess:
        lms = lms[:, ::-1]

        mask, diff = self.process(mask, lms, device=self.device)
-        image = image.resize((self.img_size, self.img_size), Image.ANTIALIAS)
+        image = image.resize((self.img_size, self.img_size), Image.Resampling.NEAREST)
        image = self.transform(image)
        real = to_var(image.unsqueeze(0))
        return [real, mask, diff], face_on_image, crop_face