facenet-pytorch icon indicating copy to clipboard operation
facenet-pytorch copied to clipboard

mtcnn detect face error

Open yueyihua opened this issue 3 years ago • 2 comments

Running on device: cpu
/root/.cache/torch
Traceback (most recent call last):
  File "mytest.py", line 33, in <module>
    x_aligned, prob = mtcnn(x, return_prob=True)
  File "/usr/local/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch/nn/modules/module.py", line 541, in __call__
    result = self.forward(*input, **kwargs)
  File "/usr/local/anaconda3/envs/pytorch/lib/python3.6/site-packages/facenet_pytorch/models/mtcnn.py", line 258, in forward
    batch_boxes, batch_probs, batch_points = self.detect(img, landmarks=True)
  File "/usr/local/anaconda3/envs/pytorch/lib/python3.6/site-packages/facenet_pytorch/models/mtcnn.py", line 317, in detect
    self.device
  File "/usr/local/anaconda3/envs/pytorch/lib/python3.6/site-packages/facenet_pytorch/models/utils/detect_face.py", line 79, in detect_face
    pick = batched_nms(boxes_scale[:, :4], boxes_scale[:, 4], image_inds_scale, 0.5)
  File "/usr/local/anaconda3/envs/pytorch/lib/python3.6/site-packages/torchvision/ops/boxes.py", line 72, in batched_nms
    keep = nms(boxes_for_nms, scores, iou_threshold)
  File "/usr/local/anaconda3/envs/pytorch/lib/python3.6/site-packages/torchvision/ops/boxes.py", line 32, in nms
    _C = _lazy_import()
  File "/usr/local/anaconda3/envs/pytorch/lib/python3.6/site-packages/torchvision/extension.py", line 12, in _lazy_import
    from torchvision import _C as C
ImportError: /usr/local/anaconda3/envs/pytorch/lib/python3.6/site-packages/torchvision/_C.cpython-36m-x86_64-linux-gnu.so: undefined symbol: _ZN3c107Warning4warnENS_14SourceLocationESs

yueyihua avatar Oct 09 '20 09:10 yueyihua

@yueyihua can you let us know what versions are you using for:

  • facenet-pytorch
  • pytorch
  • torchvision
  • python

timesler avatar Oct 09 '20 15:10 timesler

@timesler the versions are as follows: facenet-pytorch: 2.5.0 pytorch: 1.3.1 torchvision: 0.4.2 python: 3.6.9

yueyihua avatar Oct 10 '20 00:10 yueyihua