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

Preload OD models before import to check quantization (#2056)

Browse Source 
Co-authored-by: samfreund <samf.236@proton.me>
Co-authored-by: Sam Freund <techguy763@gmail.com>
This commit is contained in:
Rikhil Chilka
2025-08-16 01:59:22 -04:00
committed by GitHub
parent 2ab7a2e389
commit 2eb224a55f
4 changed files with 49 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
photon-core/src/main/java/org/photonvision/vision/objects/RknnObjectDetector.java
View File

@@ -74,6 +74,8 @@ public class RknnObjectDetector implements ObjectDetector {
if (objPointer <= 0) {
throw new RuntimeException(
"Failed to create detector from path " + model.modelFile.getPath());
} else if (!RknnJNI.isQuantized(objPointer)) {
throw new UnsupportedOperationException("Model must be quantized.");
}
logger.debug("Created detector for model " + model.modelFile.getName());

2
photon-core/src/main/java/org/photonvision/vision/objects/RubikObjectDetector.java
View File

@@ -80,6 +80,8 @@ public class RubikObjectDetector implements ObjectDetector {
+ ". Please ensure the model is valid and compatible with the Rubik backend.");
throw new RuntimeException(
"Failed to create detector from path " + model.modelFile.getPath());
} else if (!RubikJNI.isQuantized(ptr)) {
throw new UnsupportedOperationException("Model must be quantized.");
}
logger.debug("Created detector for model " + model.modelFile.getName());