search

1adrianb / face-alignment

:fire: 2D and 3D Face alignment library build using pytorch
https://www.adrianbulat.com
BSD 3-Clause "New" or "Revised" License
7.03k stars 1.34k forks source link

running the example but get error. #257

Open timeanddoctor opened 3 years ago

timeanddoctor commented 3 years ago

File "d:\ProgramData\Anaconda3\lib\site-packages\face_alignment\api.py", line 106, in get_landmarks return self.get_landmarks_from_image(image_or_path, detected_faces)

File "d:\ProgramData\Anaconda3\lib\site-packages\torch\autograd\grad_mode.py", line 26, in decorate_context return func(*args, **kwargs)

File "d:\ProgramData\Anaconda3\lib\site-packages\face_alignment\api.py", line 150, in get_landmarks_from_image pts, pts_img = get_preds_fromhm(out, center.numpy(), scale)

File "d:\ProgramData\Anaconda3\lib\site-packages\face_alignment\utils.py", line 199, in get_preds_fromhm preds, preds_orig = _get_preds_fromhm(hm, idx, center, scale)

File "d:\ProgramData\Anaconda3\lib\site-packages\numba\dispatcher.py", line 344, in _compile_for_args reraise(type(e), e, None)

File "d:\ProgramData\Anaconda3\lib\site-packages\numba\six.py", line 658, in reraise raise value.with_traceback(tb)

TypingError: Unknown attribute 'repeat' of type array(int64, 2d, C)

File "d:\ProgramData\Anaconda3\lib\site-packages\face_alignment\utils.py", line 220: def _get_preds_fromhm(hm, idx, center=None, scale=None):

idx += 1
preds = idx.repeat(2).reshape(B, C, 2).astype(np.float32)
^

[1] During: typing of get attribute at d:\ProgramData\Anaconda3\lib\site-packages\face_alignment\utils.py (220)

timeanddoctor commented 3 years ago

import face_alignment from skimage import io fa = face_alignment.FaceAlignment(face_alignment.LandmarksType._2D, device='cpu') input = io.imread('1.jpg') preds = fa.get_landmarks(input)

1adrianb commented 3 years ago

I just run the .py file from the examples folder and I was unable to reproduce your error, neither on a windows nor linux machine. Please make sure that you are using an updated numpy and numba package.