stuebinm
ff1a94e523
The nixpkgs-unstable channel's programs.sqlite was used to identify packages producing exactly one binary, and these automatically added to their package definitions wherever possible.
79 lines
1.4 KiB
Nix
79 lines
1.4 KiB
Nix
{ lib
|
|
, albumentations
|
|
, buildPythonPackage
|
|
, cython
|
|
, easydict
|
|
, fetchPypi
|
|
, insightface
|
|
, matplotlib
|
|
, mxnet
|
|
, numpy
|
|
, onnx
|
|
, onnxruntime
|
|
, opencv4
|
|
, prettytable
|
|
, pythonOlder
|
|
, scikit-image
|
|
, scikit-learn
|
|
, tensorboard
|
|
, testers
|
|
, tqdm
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "insightface";
|
|
version = "0.7.3";
|
|
format = "setuptools";
|
|
|
|
disabled = pythonOlder "3.8";
|
|
|
|
src = fetchPypi {
|
|
inherit pname version;
|
|
hash = "sha256-8ZH3GWEuuzcBj0GTaBRQBUTND4bm/NZ2wCPzVMZo3fc=";
|
|
};
|
|
|
|
nativeBuildInputs = [
|
|
cython
|
|
];
|
|
|
|
propagatedBuildInputs = [
|
|
easydict
|
|
matplotlib
|
|
mxnet
|
|
numpy
|
|
onnx
|
|
onnxruntime
|
|
opencv4
|
|
scikit-learn
|
|
scikit-image
|
|
tensorboard
|
|
tqdm
|
|
albumentations
|
|
prettytable
|
|
];
|
|
|
|
pythonImportsCheck = [
|
|
"insightface"
|
|
"insightface.app"
|
|
"insightface.data"
|
|
];
|
|
|
|
passthru.tests.version = testers.testVersion {
|
|
package = insightface;
|
|
command = "insightface-cli --help";
|
|
# Doesn't support --version but we still want to make sure the cli is executable
|
|
# and returns the help output
|
|
version = "help";
|
|
};
|
|
|
|
doCheck = false; # Upstream has no tests
|
|
|
|
meta = with lib; {
|
|
description = "State-of-the-art 2D and 3D Face Analysis Project";
|
|
mainProgram = "insightface-cli";
|
|
homepage = "https://github.com/deepinsight/insightface";
|
|
license = licenses.mit;
|
|
maintainers = with maintainers; [ oddlama ];
|
|
};
|
|
}
|