timesler / facenet-pytorch

Pretrained Pytorch face detection (MTCNN) and facial recognition (InceptionResnet) models
MIT License
4.54k stars 951 forks source link

ONNX conversion #200

Open Fritskee opened 2 years ago

Fritskee commented 2 years ago

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
beybars1 commented 1 year ago

Any updates?

chaitanyakkumar commented 1 year ago

Please update if it has been solved

beybars1 commented 1 year ago

You need to use a real image to convert the model (preferably a face), and this error will be reported if you use the data generated by torch.randn

He is right.

sceddd commented 3 months ago

so in case if the model is not good enough to detect a face it will probably throw this error?