×

签到

分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

yolov3后处理问题 已完结 止战之殇2021-11-12 11:53:09 回复 1 查看 使用求助
yolov3后处理问题
分享到:

vector<vector<vector<float>>> getResults(float* outputData,
                                         int dimNumm,
                                         int *dimValues) {
  vector<vector<vector<float>>> detections;
 
  // BangOp implementation
  float max_limit = 1;
  float min_limit = 0;
  int batchSize = dimValues[0];
  int count = dimValues[3];
  for (int i = 0; i < batchSize; i++) {
    int num_boxes = static_cast<int>(outputData[i * count]);
    vector<vector<float>> batch_box;
    for (int k = 0; k < num_boxes; k++) {
      int index = i * count + 64 + k * 7;
      vector<float> single_box;
      float bl = std::max(
          min_limit, std::min(max_limit, outputData[index + 3]));  // x1
      float br = std::max(
          min_limit, std::min(max_limit, outputData[index + 5]));  // x2
      float bt = std::max(
          min_limit, std::min(max_limit, outputData[index + 4]));  // y1
       SyntaxHighlighter.all();
版权所有 © 2024 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1