概率指数yolov2 darknet openCV3.4

Question

我是指https://github.com/opencv/opencv/blob/master/samples/dnn/yolo_object_detection.cpp

第136行是什么意思？

const int probability_index = 5;

如何修改边界框计算我只是对图像进行分类（我对目标检测感兴趣但只是图像分类）？

Answer 1

YOLO模型，至少YoloV2voc和TinyYoloV2voc，生成形状Nx(C+4)的输出矩阵，其中N是一些检测，C是一个数字类（包括背景），4是向量[centerX, centerY, width, height]。所以分类置信度从第5个元素开始。