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