implement individual export

Author: samfreund

photon-client/src/components/settings/ObjectDetectionCard.vue

@@ -170,6 +170,11 @@ const openExportPrompt = () => {
   exportModels.value.click();
 };
 
+const exportIndividualModel = ref();
+const openExportIndividualModelPrompt = () => {
+  exportIndividualModel.value.click();
+};
+
 const showNukeDialog = ref(false);
 const expected = "Delete Models";
 const yesDeleteMyModelsText = ref("");
@@ -367,6 +372,17 @@ const nukeModels = () => {
           <v-dialog v-model="showInfo.show" width="600">
             <v-card color="primary" dark>
               <v-card-title>Object Detection Model Info</v-card-title>
+              <v-btn color="secondary" @click="openExportIndividualModelPrompt">
+                <v-icon left class="open-icon"> mdi-export </v-icon>
+                <span class="open-label">Export Models</span>
+              </v-btn>
+              <a
+                ref="exportIndividualModel"
+                style="color: black; text-decoration: none; display: none"
+                :href="`http://${address}/api/objectdetection/exportIndividual?modelPath=${showInfo.model.modelPath}`"
+                :download="`${showInfo.model.nickname}-${showInfo.model.family}-${showInfo.model.version}-${showInfo.model.resolutionWidth}x${showInfo.model.resolutionHeight}.zip`"
+                target="_blank"
+              />
               <v-card-text>
                 <p>Model Path: {{ showInfo.model.modelPath }}</p>
                 <p>Model Nickname: {{ showInfo.model.nickname }}</p>

photon-core/src/main/java/org/photonvision/common/configuration/NeuralNetworkModelManager.java

@@ -39,6 +39,7 @@
 import org.photonvision.common.logging.Logger;
 import org.photonvision.vision.objects.Model;
 import org.photonvision.vision.objects.RknnModel;
+import org.zeroturnaround.zip.ZipUtil;
 
 /**
  * Manages the loading of neural network models.
@@ -331,4 +332,42 @@ public boolean nukeModels() {
         // Delete model info
         return ConfigManager.getInstance().getConfig().neuralNetworkPropertyManager().nuke();
     }
+
+    public File zipModel(String modelPath) {
+        File modelFile = new File(modelPath);
+        if (!modelFile.exists()) {
+            logger.error("Model file does not exist: " + modelFile.getAbsolutePath());
+            return null;
+        }
+
+        ModelProperties properties =
+                ConfigManager.getInstance()
+                        .getConfig()
+                        .neuralNetworkPropertyManager()
+                        .getModel(Path.of(modelPath));
+
+        String fileName = "";
+        if (properties != null) {
+            fileName =
+                    String.format(
+                            "%s-%s-%s-%dx%d",
+                            properties.nickname().replace(" ", ""),
+                            properties.family(),
+                            properties.version(),
+                            properties.resolutionWidth(),
+                            properties.resolutionHeight());
+        } else {
+            fileName = new File(modelPath).getName();
+        }
+
+        try {
+            var out = Files.createTempFile(fileName, ".zip");
+            ZipUtil.pack(new File(modelPath), out.toFile());
+            return out.toFile();
+        } catch (Exception e) {
+            e.printStackTrace();
+            logger.error("Failed to zip model file: " + modelFile.getAbsolutePath(), e);
+            return null;
+        }
+    }
 }

photon-server/src/main/java/org/photonvision/server/RequestHandler.java

@@ -682,6 +682,37 @@ public static void onExportObjectDetectionModelRequest(Context ctx) {
         }
     }
 
+    public static void onExportIndividualObjectDetectionModelRequest(Context ctx) {
+        logger.info("Exporting Object Detection Model to ZIP Archive");
+
+        try {
+            var modelPath = ctx.formParam("modelPath");
+
+            if (modelPath == null || modelPath.isEmpty()) {
+                ctx.status(400);
+                ctx.result("The provided model path was malformed");
+                logger.error("The provided model path was malformed");
+                return;
+            }
+
+            var zip = NeuralNetworkModelManager.getInstance().zipModel(modelPath);
+
+            var stream = new FileInputStream(zip);
+            logger.info("Uploading object detection model with size " + stream.available());
+
+            ctx.contentType("application/zip");
+            ctx.header("Content-Disposition", "attachment; filename=" + zip.getName());
+
+            ctx.result(stream);
+            ctx.status(200);
+        } catch (IOException e) {
+            logger.error(
+                    "Unable to export object detection model archive, bad recode from zip to byte, " + e);
+            ctx.status(500);
+            ctx.result("There was an error while exporting the object detection model archive");
+        }
+    }
+
     public static void onBulkImportObjectDetectionModelRequest(Context ctx) {
         var file = ctx.uploadedFile("data");
 

photon-server/src/main/java/org/photonvision/server/Server.java

@@ -150,6 +150,9 @@ private static void start(int port) {
         // Object detection
         app.post("/api/objectdetection/import", RequestHandler::onImportObjectDetectionModelRequest);
         app.post("/api/objectdetection/export", RequestHandler::onExportObjectDetectionModelRequest);
+        app.post(
+                "/api/objectdetection/exportindividual",
+                RequestHandler::onExportIndividualObjectDetectionModelRequest);
         app.post("/api/objectdetection/delete", RequestHandler::onDeleteObjectDetectionModelRequest);
         app.post("/api/objectdetection/rename", RequestHandler::onRenameObjectDetectionModelRequest);
         app.post("/api/objectdetection/nuke", RequestHandler::onNukeObjectDetectionModelsRequest);