Added a functional ultralytics-thop
This commit is contained in:
1 parent
72156eb50a
commit
aa51433d91
1 file changed
+4
-2
@@ -40,7 +40,7 @@
|
|||||||
};
|
};
|
||||||
};
|
};
|
||||||
|
|
||||||
python-path = pkgs.python3Packages.buildPythonPackage rec {
|
ultralytics = pkgs.python3Packages.buildPythonPackage rec {
|
||||||
# pname = "opencv-python";
|
# pname = "opencv-python";
|
||||||
# version = "4.10.0.84";
|
# version = "4.10.0.84";
|
||||||
|
|
||||||
@@ -64,6 +64,8 @@
|
|||||||
pythonPackages.torchvision
|
pythonPackages.torchvision
|
||||||
pythonPackages.pandas
|
pythonPackages.pandas
|
||||||
pythonPackages.matplotlib
|
pythonPackages.matplotlib
|
||||||
|
pythonPackages.seaborn
|
||||||
|
ultralytics-thop
|
||||||
|
|
||||||
];
|
];
|
||||||
|
|
||||||
@@ -97,7 +99,7 @@
|
|||||||
};
|
};
|
||||||
};
|
};
|
||||||
|
|
||||||
env = pkgs.python312.withPackages (ps: [ ultralytics-thop ]);
|
env = pkgs.python312.withPackages (ps: [ ultralytics ]);
|
||||||
|
|
||||||
in {
|
in {
|
||||||
packages = {
|
packages = {
|
||||||
|
|||||||
Reference in new issue
Block a user