tjh.dev
1{
2 buildPythonPackage,
3 fetchPypi,
4 pillow,
5 torchvision,
6 lib,
7}:
8
9buildPythonPackage rec {
10 pname = "facenet-pytorch";
11 version = "2.5.3";
12
13 src = fetchPypi {
14 inherit pname version;
15 hash = "sha256-mMxbQqSPg3wCPrkvKlcc1KxqRmh8XnG56ZtJEIcnPis=";
16 };
17
18 doCheck = false; # pypi version doesn't ship with tests
19
20 pythonImportsCheck = [ "facenet_pytorch" ];
21
22 propagatedBuildInputs = [
23 pillow
24 torchvision
25 ];
26
27 meta = {
28 description = "Pretrained Pytorch face detection (MTCNN) and facial recognition (InceptionResnet) models";
29 homepage = "https://github.com/timesler/facenet-pytorch";
30 license = lib.licenses.mit;
31 maintainers = [ lib.maintainers.lucasew ];
32 };
33}