public/objectifier
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Need to pass yolo-data to module

Browse Source
This commit is contained in:
niten 2023-01-06 17:31:58 -08:00
parent 7eef72712c
commit 31d032b5ef
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
flake.nix
View File

@ -52,7 +52,7 @@
inherit (self.packages."${prev.system}") objectifier yolo-cli;
};
objectifier = final: prev: {
inherit (self.packages."${prev.system}") objectifier;
inherit (self.packages."${prev.system}") objectifier yolo-data;
};
yolo-cli = final: prev: {
inherit (self.packages."${prev.system}") yolo-cli;

8
objectifier-module.nix
View File

@ -1,3 +1,5 @@
{ config, lib, pkgs, ... }:
with lib;
@ -44,9 +46,9 @@ in {
reloadIfChanged = true;
path = with pkgs; [ pythonYolo ];
environment = {
OBJECTIFIER_YOLOV3_CONFIG = "${yolo-data}/yolov3.cfg";
OBJECTIFIER_YOLOV3_WEIGHTS = "${yolo-data}/yolov3.weights";
OBJECTIFIER_YOLOV3_LABELS = "${yolo-data}/labels";
OBJECTIFIER_YOLOV3_CONFIG = "${pkgs.yolo-data}/yolov3.cfg";
OBJECTIFIER_YOLOV3_WEIGHTS = "${pkgs.yolo-data}/yolov3.weights";
OBJECTIFIER_YOLOV3_LABELS = "${pkgs.yolo-data}/labels";
OBJECTIFIER_BUFFER_SIZE = 524288;
};
serviceConfig = {