I am going to extract the face landmark using python FaceAlignment package.
My OS is windows10 and i installed torch 1.4.0 and torchvision 0.5 and cuda version 10.1.
I can get the face landmark with square image using get_landmarks() function.
But when i use the not square image, i can’t get the result.
Instead i get error as bellow:
Traceback (most recent call last):
File “.\detlandmark\detlandmark.py”, line 12, in 
preds = fa.get_landmarks(input)
File “C:\Users\user\Anaconda3\lib\site-packages\face_alignment\api.py”, line 141, in get_landmarks
return self.get_landmarks_from_image(image_or_path, detected_faces)
File “C:\Users\user\Anaconda3\lib\site-packages\face_alignment\api.py”, line 171, in get_landmarks_from_image
detected_faces = self.face_detector.detect_from_image(image[…, ::-1].copy())
File “C:\Users\user\Anaconda3\lib\site-packages\face_alignment\detection\sfd\sfd_detector.py”, line 51, in detect_from_image
bboxlist = detect(self.face_detector, image, device=self.device)
File “C:\Users\user\Anaconda3\lib\site-packages\face_alignment\detection\sfd\detect.py”, line 30, in detect
olist = net(img)
File “C:\Users\user\Anaconda3\lib\site-packages\torch\nn\modules\module.py”, line 532, in call
result = self.forward(*input, **kwargs)
File “C:\Users\user\Anaconda3\lib\site-packages\face_alignment\detection\sfd\net_s3fd.py”, line 80, in forward
h = F.relu(self.conv3_2(h))
File “C:\Users\user\Anaconda3\lib\site-packages\torch\nn\modules\module.py”, line 532, in call
result = self.forward(*input, **kwargs)
File “C:\Users\user\Anaconda3\lib\site-packages\torch\nn\modules\conv.py”, line 345, in forward
return self.conv2d_forward(input, self.weight)
File “C:\Users\user\Anaconda3\lib\site-packages\torch\nn\modules\conv.py”, line 342, in conv2d_forward
self.padding, self.dilation, self.groups)
RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR
I want know the origin of my error.
Please help me~!!!