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

Add FRC 2026 game piece detection models (#2332)

Browse Source 
Added a model for detecting Rebuilt's FUEL game piece for versions of
the Orange Pi 5 and the Rubik Pi 3, which lowers the bar for less
experienced FRC teams to use object detection on their robots. Also
updated the documentation of object detection to reference the new
models.

The original model weights were trained by FRC team 2826 and were
published on this [Chief Delphi
thread](https://www.chiefdelphi.com/t/introducing-wave-robotics-yolov11-model-for-rebuilt/512701).

The dataset used for RKNN quantization is the [Fuel V3
dataset](https://universe.roboflow.com/frcroboraiders/frc-2026-fuel-sbrdk/dataset/3)
provided by
[frcroboraiders](https://universe.roboflow.com/frcroboraiders) on
Roboflow Universe.
This commit is contained in:
Cooper Bouchard
2026-01-28 00:47:54 -05:00
committed by GitHub
parent 49629afe9b
commit 7cc22e52ea
6 changed files with 16 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -149,16 +149,6 @@ public class NeuralNetworkModelManager {
"hair drier", // Typo in official COCO documentation
"toothbrush"));
nnProps.addModelProperties(
new ModelProperties(
Path.of(modelsDirectory.getAbsolutePath(), "algaeV1-640-640-yolov8n.rknn"),
"Algae v8n",
new LinkedList<String>(List.of("Algae")),
640,
480,
Family.RKNN,
Version.YOLOV8));
nnProps.addModelProperties(
new ModelProperties(
Path.of(modelsDirectory.getAbsolutePath(), "yolov8nCOCO.rknn"),
@@ -171,13 +161,13 @@ public class NeuralNetworkModelManager {
nnProps.addModelProperties(
new ModelProperties(
Path.of(modelsDirectory.getAbsolutePath(), "algae-coral-yolov8s.tflite"),
"Algae Coral v8s",
new LinkedList<String>(List.of("Algae", "Coral")),
Path.of(modelsDirectory.getAbsolutePath(), "fuelV1-yolo11n.rknn"),
"Fuel v11n",
new LinkedList<String>(List.of("Fuel")),
640,
640,
Family.RUBIK,
Version.YOLOV8));
Family.RKNN,
Version.YOLOV11));
nnProps.addModelProperties(
new ModelProperties(
@@ -189,6 +179,16 @@ public class NeuralNetworkModelManager {
Family.RUBIK,
Version.YOLOV8));
nnProps.addModelProperties(
new ModelProperties(
Path.of(modelsDirectory.getAbsolutePath(), "fuelV1-yolo11n.tflite"),
"Fuel v11n",
new LinkedList<String>(List.of("Fuel")),
640,
640,
Family.RUBIK,
Version.YOLOV11));
return nnProps;
}