浏览代码

1.准备进行AI运行测试

kappo 2 月之前
父节点
当前提交
54431b1b78
共有 2 个文件被更改,包括 8 次插入1 次删除
  1. 6 1
      AIManager/RKNNManager.cpp
  2. 2 0
      main.cpp

+ 6 - 1
AIManager/RKNNManager.cpp

@@ -28,16 +28,21 @@ void RKNNManager::taskThread(std::string modelpath)
     DataManager::getInstance().addDataPipe<DataPackage>(pipeName);
 
     //time test
-    ImageTest::saveImageFromData
+    ImageTest::HighResolutionTimer timer;
 
     while (m_threadSwitch)
     {
         if (DataManager::getInstance().popData("resized" + std::to_string(index), dataPackage))
         {
             
+            timer.start();
 
             infer.infer(index, (unsigned char *)dataPackage->pResizeData, dataPackage->nResizeWidth, dataPackage->nResizeHeight);
 
+            timer.stop();
+
+            std::cout << "infer time: " << timer.elapsed_milliseconds() << " ms" << std::endl;
+
             continue;
         }
 

+ 2 - 0
main.cpp

@@ -20,6 +20,8 @@ int main(int argc, char **argv)
 
     MppManager::getInstance().addMppDecode();
 
+    RKNNManager::getInstance().addRknnTask("../ppyoloe.rknn");
+
     // Start all streaming
     UVCManager::getInstance().startAllStreaming();