Need to pass yolo-data to module

This commit is contained in:
2023-01-06 17:31:58 -08:00
parent 7eef72712c
commit 31d032b5ef
2 changed files with 6 additions and 4 deletions
+5 -3
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 = {