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