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";
|
||||
# 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 = {
|
||||
|
||||
Reference in new issue
Block a user