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

ONNX conversion

Open Fritskee opened this issue 1 year ago • 4 comments

Has anybody successfully converted the MTCNN to ONNX? Keep getting the error:

~\Anaconda3\envs\facenet\lib\site-packages\facenet_pytorch\models\utils\detect_face.py in detect_face(imgs, minsize, pnet, rnet, onet, threshold, factor, device)
     81         offset += boxes_scale.shape[0]
     82 
---> 83     boxes = torch.cat(boxes, dim=0)
     84     image_inds = torch.cat(image_inds, dim=0)
     85 

RuntimeError: torch.cat(): expected a non-empty list of Tensors

Fritskee avatar Sep 27 '22 15:09 Fritskee