pyrox.dev
1{ lib
2, buildPythonPackage
3, fetchPypi
4# build inputs
5, torch
6, torchvision
7, timm
8, pycocotools
9, omegaconf
10}:
11let
12 pname = "effdet";
13 version = "0.4.1";
14in
15buildPythonPackage {
16 inherit pname version;
17 format = "setuptools";
18
19 src = fetchPypi {
20 inherit pname version;
21 hash = "sha256-rFWJ/TBKVlDCAZhrLvX44QwREJOnGxxJ+muIF3EIErU=";
22 };
23
24 propagatedBuildInputs = [
25 torch
26 torchvision
27 timm
28 pycocotools
29 omegaconf
30 ];
31
32 # Project has no tests
33 doCheck = false;
34
35 pythonImportsCheck = [ "effdet" ];
36
37 meta = {
38 description = "A PyTorch implementation of EfficientDet";
39 homepage = "https://pypi.org/project/effdet";
40 license = lib.licenses.asl20;
41 maintainers = with lib.maintainers; [ happysalada ];
42 };
43}