added support for all possible core masks

PhotonVision · Jan 29, 2024 · ecc58ce · ecc58ce
1 parent 10e5403
commit ecc58ce
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/main/native/cpp/yolo_common.cpp b/src/main/native/cpp/yolo_common.cpp
@@ -95,6 +95,12 @@ YoloModel::YoloModel(std::string modelPath, int num_classes_, ModelVersion type_
     case 2:
         core_mask = RKNN_NPU_CORE_2;
         break;
+    case 10:
+        core_mask = RKNN_NPU_CORE_0_1;
+        break;
+    case 210:
+        core_mask = RKNN_NPU_CORE_0_1_2;
+        break;
     default:
         core_mask = RKNN_NPU_CORE_AUTO;
         break;