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";
# version = "4.10.0.84";
@@ -64,6 +64,8 @@
pythonPackages.torchvision
pythonPackages.pandas
pythonPackages.matplotlib
pythonPackages.seaborn
ultralytics-thop
];
@@ -97,7 +99,7 @@
};
};
env = pkgs.python312.withPackages (ps: [ ultralytics-thop ]);
env = pkgs.python312.withPackages (ps: [ ultralytics ]);
in {
packages = {