objectifier/yolo-cli.nix

{ pkgs, yolov3-data, ... }:

let
  name = "yolo-cli";
  pythonYolo = pkgs.python3.withPackages (pyPkgs: with pyPkgs; [ opencv4 ]);
  yoloCliFiles = pkgs.stdenv.mkDerivation {
    name = "yolo-cli-src";
    src = ./src;
    phases = [ "installPhase" ];
    installPhase = ''
      mkdir -p $out
      cp $src/detector.py $out/detector.py
      cp $src/yolo-cli.py $out/yolo-cli.py
      chmod +x $out/yolo-cli.py
    '';
  };

in pkgs.writeShellApplication {
  inherit name;
  runtimeInputs = [ pythonYolo ];
  text = pkgs.lib.concatStringsSep " " [
    "${yoloCliFiles}/yolo-cli.py"
    "--yolo_weights=${yolov3-data}/yolov3.weights"
    "--yolo_config=${yolov3-data}/yolov3.cfg"
    "--yolo_labels=${yolov3-data}/labels"
    ''"$@"''
  ];
}