2025-07-24 13:55:40 +02:00

57 lines
980 B
Nix

{
lib,
buildPythonPackage,
fetchFromGitHub,
# build-system
setuptools,
# dependencies
joblib,
keras,
lz4,
pythonAtLeast,
distutils,
# tests
pytestCheckHook,
}:
buildPythonPackage rec {
pname = "mtcnn";
version = "1.0.0";
pyproject = true;
src = fetchFromGitHub {
owner = "ipazc";
repo = "mtcnn";
tag = "v${version}";
hash = "sha256-gp+jfa1arD3PpJpuRFKIUznV0Lyjt3DPn/HHUviDXhk=";
};
build-system = [ setuptools ];
dependencies = [
joblib
lz4
]
++ lib.optionals (pythonAtLeast "3.12") [
distutils
];
pythonImportsCheck = [ "mtcnn" ];
nativeCheckInputs = [
keras
pytestCheckHook
];
meta = {
description = "MTCNN face detection implementation for TensorFlow";
homepage = "https://github.com/ipazc/mtcnn";
changelog = "https://github.com/ipazc/mtcnn/releases/tag/v${version}";
license = lib.licenses.mit;
maintainers = with lib.maintainers; [ derdennisop ];
};
}