diff --git a/models/utils/detect_face.py b/models/utils/detect_face.py index f5db6a40..ff5d514d 100644 --- a/models/utils/detect_face.py +++ b/models/utils/detect_face.py @@ -35,7 +35,7 @@ def detect_face(imgs, minsize, pnet, rnet, onet, threshold, factor, device): else: if not isinstance(imgs, (list, tuple)): imgs = [imgs] - if any(img.size != imgs[0].size for img in imgs): + if any(img.size() != imgs[0].size() for img in imgs): raise Exception("MTCNN batch processing only compatible with equal-dimension images.") imgs = np.stack([np.uint8(img) for img in imgs]) imgs = torch.as_tensor(imgs.copy(), device=device)