Added a functional ultralytics-thop

This commit is contained in:
didier committed 2024-10-14 14:16:56 -04:00
1 parent 72156eb50a
commit aa51433d91
1 file changed
+4 -2
+4 -2
View File
@@ -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 = {