std::vector<ObjDetect_Results> parseModelOutput(float output, int num_classes = 7, float confidence_threshold = 0.55f) { std::vector<ObjDetect_Results> detections; float data = output; const int num_boxes = 8400; const int attributes_per_box = 4 + num_classes; // 4坐标 + 1置信度 + n类别 // for (int w = 0; w < 8400; w++) // { // for (int h = 0; h < attributes_per_box; h++) // { // if (h > 3) // { // printf(" %.2f ",sigmoid(1-data[h 8400 + w])); // } // else // printf(" %.2f ",data[h 8400 + w]); // } // printf("\r\n"); // } // return detections;
