pkgs/development/python-modules/imantics/default.nix

   1 { buildPythonPackage
   2 , fetchFromGitHub
   3 , lib
   4 , numpy
   5 , opencv4
   6 , lxml
   7 , xmljson
   8 , pytestCheckHook
   9 }:
  10
  11 buildPythonPackage rec {
  12   pname = "imantics";
  13   version = "0.1.12";
  14
  15   src = fetchFromGitHub {
  16     owner = "jsbroks";
  17     repo = "imantics";
  18     rev = "76d81036d8f92854d63ad9938dd76c718f8b482e";
  19     sha256 = "1zv2gj8cbakhh2fyr2611cbqhfk37a56x973ny9n43y70n26pzm8";
  20   };
  21
  22   propagatedBuildInputs = [
  23     numpy
  24     opencv4
  25     lxml
  26     xmljson
  27   ];
  28
  29   postPatch = ''
  30     substituteInPlace setup.py \
  31       --replace "'opencv-python>=3'," ""
  32   '';
  33
  34   nativeCheckInputs = [ pytestCheckHook ];
  35
  36   pythonImportsCheck = [ "imantics" ];
  37
  38   meta = with lib; {
  39     description = "Convert and visualize many annotation formats for object dectection and localization";
  40     homepage = "https://github.com/jsbroks/imantics";
  41     license = with licenses; [ mit ];
  42     maintainers = [ maintainers.rakesh4g ];
  43   };
  44 }