facenet-pytorch
facenet-pytorch copied to clipboard
timesler
Pretrained Pytorch face detection (MTCNN) and facial recognition (InceptionResnet) models
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...
Hello, I have a problem with memory that keeps increasing after face detection on about 100 images. Is there a solution or workaround to the problem? The details are as...
Previously I was working on the https://github.com/davidsandberg/facenet, but there is a limitation where the repo is base on tensorflow 1.x and latest hardware for Geforce 30 series and above is...
Problem: muti error occurred when I try to run the example from `help(MTCNN)` on my MacBook Air M1. Env: `macOS Monterey 12.6` `MacBook Air(M1, 2020)` `8GB RAM` `iBoot 7459.141.1` `conda...
bug-fix` .size `instead of `size()`. `size` is the method. `size()` is the correct equivalent to `shape`
> venv/lib/python3.12/site-packages/facenet_pytorch/models/mtcnn.py", line 444, in select_boxes > selected_boxes = np.array(selected_boxes) > ^^^^^^^^^^^^^^^^^^^^^^^^ > ValueError: setting an array element with a sequence. The requested array has an inhomogeneous shape after 1...
I am running Python 3.9 on Apple M1. At the beginning of a script - ``` from facenet_pytorch import MTCNN import torch device = torch.device('mps') mtcnn = MTCNN(device=device) ``` led...
Hello ! I've just stumbled upon facenet-pytorch. according to the documentation, this module can detect and recognize faces. In my case, I'd like to classifying 20,000 images based on the...
Hello! Awesome lib! In the current version of the code, I get a `TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host...
Hi, Does anyone know of a method to retrain the model with some images and then test it with images of the same and different people? I mean, it is...
