revert model extractions

PhotonVision · Sep 19, 2024 · 2e6265e · 2e6265e
1 parent 4836e63
commit 2e6265e
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/...n-core/src/main/java/org/photonvision/common/configuration/NeuralNetworkModelManager.java b/...n-core/src/main/java/org/photonvision/common/configuration/NeuralNetworkModelManager.java
@@ -177,6 +177,7 @@ public Optional<Model> getDefaultModel() {
         if (supportedBackends.isEmpty()) {
             return Optional.empty();
         }
+
         return models.get(supportedBackends.get(0)).stream().findFirst();
     }
 
@@ -288,10 +289,17 @@ public void extractModels(File modelsDirectory) {
                     Path outputPath =
                             modelsDirectory.toPath().resolve(entry.getName().substring(resource.length() + 1));
 
-                    logger.info("Extracting DNN resource: " + entry.getName());
+                    if (Files.exists(outputPath)) {
+                        logger.info("Skipping extraction of DNN resource: " + entry.getName());
+                        continue;
+                    }
+
                     Files.createDirectories(outputPath.getParent());
                     try (InputStream inputStream = jarFile.getInputStream(entry)) {
                         Files.copy(inputStream, outputPath, StandardCopyOption.REPLACE_EXISTING);
+                        logger.info("Extracted DNN resource: " + entry.getName());
+                    } catch (IOException e) {
+                        logger.error("Failed to extract DNN resource: " + entry.getName(), e);
                     }
                 }
             }