initial commit
This commit is contained in:
commit
6f735a22d7
12 changed files with 356 additions and 0 deletions
40
flake.nix
Normal file
40
flake.nix
Normal file
|
|
@ -0,0 +1,40 @@
|
|||
{
|
||||
description = "People Counter with YOLOv8, ByteTrack and Streamlit";
|
||||
|
||||
inputs = {
|
||||
nixpkgs.url = "github:NixOS/nixpkgs/nixos-25.05";
|
||||
flake-utils.url = "github:numtide/flake-utils";
|
||||
};
|
||||
|
||||
outputs = { nixpkgs, flake-utils, ... }:
|
||||
flake-utils.lib.eachDefaultSystem (system:
|
||||
let
|
||||
pkgs = nixpkgs.legacyPackages.${system};
|
||||
|
||||
pythonEnv = pkgs.python313.withPackages (ps: with ps; [
|
||||
numpy
|
||||
opencv-python
|
||||
streamlit
|
||||
pyyaml
|
||||
pillow
|
||||
ultralytics
|
||||
torch
|
||||
torchvision
|
||||
prometheus-client
|
||||
]);
|
||||
in {
|
||||
devShell = pkgs.mkShell {
|
||||
buildInputs = [
|
||||
pythonEnv
|
||||
pkgs.ffmpeg
|
||||
pkgs.mediamtx
|
||||
];
|
||||
|
||||
shellHook = ''
|
||||
export PYTHONUNBUFFERED=1
|
||||
export MTX_PATHS_ALL_SOURCE=publisher
|
||||
echo "People Counter DevShell ready"
|
||||
'';
|
||||
};
|
||||
});
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue