Initial commit

2023-01-06 14:46:11 -08:00
commit e240b33b77
14 changed files with 617 additions and 0 deletions
@@ -0,0 +1,37 @@
+{ pkgs, lib, buildEnv, stdenv, darknet, ... }:
+
+with lib;
+buildEnv {
+  name = "yolov3-data";
+  paths = let
+    cfg = stdenv.mkDerivation {
+      name = "yolov3-cfg";
+      src = darknet;
+      phases = [ "installPhase" ];
+      installPhase = ''
+        mkdir -p $out
+        cp $src/cfg/yolov3.cfg $out/yolov3.cfg
+      '';
+    };
+
+    labels = stdenv.mkDerivation {
+      name = "yolov3-labels";
+      src = darknet;
+      phases = [ "installPhase" ];
+      installPhase = ''
+        mkdir -p $out
+        cp $src/data/coco.names $out/labels
+      '';
+    };
+
+    weights = stdenv.mkDerivation {
+      name = "yolov3-weights";
+      src = ./data;
+      phases = [ "installPhase" ];
+      installPhase = ''
+        mkdir -p $out
+        cp $src/yolov3.weights $out/yolov3.weights
+      '';
+    };
+  in [ cfg labels weights ];
+}