cpp oop skill issues

PhotonVision · Jan 31, 2024 · 2db8b1b · 2db8b1b
1 parent 954dc1b
commit 2db8b1b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/main/native/cpp/yolo_common.cpp b/src/main/native/cpp/yolo_common.cpp
@@ -202,7 +202,7 @@ int YoloModel::changeCoreMask(int coreNumber) {
         throw std::runtime_error("invalid core selection! core selected: " + coreNumber);
         break;
     }
-    ret = rknn_set_core_mask(ctx, core_mask);
+    int ret = rknn_set_core_mask(ctx, core_mask);
     if (ret < 0)
     {
         throw std::runtime_error("rknn_init core error ret=" + ret);

diff --git a/src/main/native/include/yolo_common.hpp b/src/main/native/include/yolo_common.hpp
@@ -19,7 +19,7 @@ class YoloModel {
 public:
     YoloModel(std::string modelPath, int num_classes_, ModelVersion type_, int coreNumber);
 
-    int YoloModel::changeCoreMask(int coreNumber);
+    int changeCoreMask(int coreNumber);
 
     detect_result_group_t forward(cv::Mat &orig_img, DetectionFilterParams params);