pyrox.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 format = "setuptools";
13
14 src = fetchPypi {
15 inherit pname version;
16 hash = "sha256-mMxbQqSPg3wCPrkvKlcc1KxqRmh8XnG56ZtJEIcnPis=";
17 };
18
19 doCheck = false; # pypi version doesn't ship with tests
20
21 pythonImportsCheck = [ "facenet_pytorch" ];
22
23 propagatedBuildInputs = [
24 pillow
25 torchvision
26 ];
27
28 meta = {
29 description = "Pretrained Pytorch face detection (MTCNN) and facial recognition (InceptionResnet) models";
30 homepage = "https://github.com/timesler/facenet-pytorch";
31 license = lib.licenses.mit;
32 maintainers = [ lib.maintainers.lucasew ];
33 };
34}