Team2890/PhotonVision
4
0
Fork 0
mirror of https://github.com/PhotonVision/photonvision synced 2026-07-03 03:01:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Exclude license from being loaded as a model (#2063)

Browse Source
This commit is contained in:
Sam Freund
2025-08-15 15:17:21 -05:00
committed by GitHub
parent 6fe96316a4
commit c7f5edc262
2 changed files with 23 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -226,8 +226,18 @@ public class NeuralNetworkModelManager {
private static final Logger logger = new Logger(NeuralNetworkModelManager.class, LogGroup.Config);
public enum Family {
RKNN,
RUBIK
RKNN(".rknn"),
RUBIK(".tflite");
private final String fileExtension;
private Family(String fileExtension) {
this.fileExtension = fileExtension;
}
public String extension() {
return fileExtension;
}
}
public enum Version {
@@ -358,7 +368,11 @@ public class NeuralNetworkModelManager {
try {
Files.walk(modelsDirectory.toPath())
.filter(Files::isRegularFile)
.forEach(path -> loadModel(path));
.filter(
path ->
supportedBackends.stream()
.anyMatch(family -> path.toString().endsWith(family.extension())))
.forEach(this::loadModel);
} catch (IOException e) {
logger.error("Failed to discover models at " + modelsDirectory.getAbsolutePath(), e);
}