Team2890/PhotonVision
4
0
Fork 0
mirror of https://github.com/PhotonVision/photonvision synced 2026-06-19 00:41:41 +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

15
photon-server/src/main/java/org/photonvision/server/RequestHandler.java
View File

@@ -605,16 +605,13 @@ public class RequestHandler {
return;
}
String modelFileExtension = "";
NeuralNetworkModelManager.Family family;
switch (Platform.getCurrentPlatform()) {
case LINUX_QCS6490:
modelFileExtension = "tflite";
family = NeuralNetworkModelManager.Family.RUBIK;
break;
case LINUX_RK3588_64:
modelFileExtension = "rknn";
family = NeuralNetworkModelManager.Family.RKNN;
break;
default:
@@ -625,19 +622,19 @@ public class RequestHandler {
}
// If adding additional platforms, check platform matches
if (!modelFile.extension().contains(modelFileExtension)) {
if (!modelFile.extension().contains(family.extension())) {
ctx.status(400);
ctx.result(
"The uploaded file was not of type '"
+ modelFileExtension
+ family.extension()
+ "'. The uploaded file should be a ."
+ modelFileExtension
+ family.extension()
+ " file.");
logger.error(
"The uploaded file was not of type '"
+ modelFileExtension
+ family.extension()
+ "'. The uploaded file should be a ."
+ modelFileExtension
+ family.extension()
+ " file.");
return;
}
@@ -665,7 +662,7 @@ public class RequestHandler {
.addModelProperties(
new ModelProperties(
modelPath,
modelFile.filename().replaceAll("." + modelFileExtension, ""),
modelFile.filename().replaceAll("." + family.extension(), ""),
labels,
width,
height,