Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.4 kB · 73 lines
Nix
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374{ lib, buildPythonPackage, pythonOlder, fetchPypi, setuptools, matplotlib, numpy, opencv-python, pillow, scikit-learn, torch, torchvision, ttach, tqdm,}:
buildPythonPackage rec { pname = "grad-cam"; version = "1.5.5"; pyproject = true;
disabled = pythonOlder "3.8";
src = fetchPypi { inherit pname version; hash = "sha256-aQxDPSJtNcicnrFwRi2yBJCcsGs5xzgeaICkm2/DcBU="; };
nativeBuildInputs = [ ];
pythonRelaxDeps = [ "torchvision" ];
build-system = [ setuptools ];
dependencies = [ matplotlib numpy opencv-python pillow scikit-learn torchvision ttach tqdm ];
# Let the user bring their own instance (as with torchmetrics) buildInputs = [ torch ];
doCheck = false; # every nontrivial test tries to download a pretrained model
pythonImportsCheck = [ "pytorch_grad_cam" "pytorch_grad_cam.metrics" "pytorch_grad_cam.metrics.cam_mult_image" "pytorch_grad_cam.metrics.road" "pytorch_grad_cam.utils" "pytorch_grad_cam.utils.image" "pytorch_grad_cam.utils.model_targets" ];
meta = { description = "Advanced AI explainability for computer vision"; homepage = "https://jacobgil.github.io/pytorch-gradcam-book"; license = lib.licenses.mit; maintainers = with lib.maintainers; [ bcdarwin ]; };}