it'll work

PhotonVision · Jan 12, 2025 · 9354864 · 9354864
1 parent 2d19908
commit 9354864
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
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
@@ -136,6 +136,11 @@ public HashMap<String, ArrayList<String>> getModels() {
         return modelMap;
     }
 
+    public void rescanModels(File modelsDirectory) {
+        models = null;
+        discoverModels(modelsDirectory);
+    }
+
     /**
      * Retrieves the model with the specified name, assuming it is available under a supported
      * backend.
@@ -243,7 +248,8 @@ public void discoverModels(File modelsFolder) {
             logger.error("Failed to discover models at " + modelsFolder.getAbsolutePath(), e);
         }
 
-        // After loading all of the models, sort them by name to ensure a consistent ordering
+        // After loading all of the models, sort them by name to ensure a consistent
+        // ordering
         models.forEach(
                 (backend, backendModels) ->
                         backendModels.sort((a, b) -> a.getName().compareTo(b.getName())));

diff --git a/photon-server/src/main/java/org/photonvision/server/RequestHandler.java b/photon-server/src/main/java/org/photonvision/server/RequestHandler.java
@@ -604,7 +604,7 @@ public static void onObjectDetectionModelImportRequest(Context ctx) {
             }
 
             NeuralNetworkModelManager.getInstance()
-                    .discoverModels(ConfigManager.getInstance().getModelsDirectory());
+                    .rescanModels(ConfigManager.getInstance().getModelsDirectory());
 
             ctx.status(200).result("Successfully uploaded object detection model");
         } catch (Exception e) {