potok/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2026-07-29 07:13:02 +04:00
Code Packages Releases Wiki Activity

bound darknet cfg layer and anchor indices before vector reads

Browse Source
This commit is contained in:
uwezkhan
2026-06-15 21:36:38 +05:30
parent 6df9732c81
commit afbfa275d8
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/dnn/src/darknet/darknet_io.cpp
+5 -2
View File
@@ -534,8 +534,10 @@ namespace cv {
std::vector<float> usedAnchors(numAnchors * 2);
for (int i = 0; i < numAnchors; ++i)
{
usedAnchors[i * 2] = anchors[mask[i] * 2];
usedAnchors[i * 2 + 1] = anchors[mask[i] * 2 + 1];
const int m = mask[i];
CV_Assert(m >= 0 && static_cast<size_t>(m) * 2 + 1 < anchors.size());
usedAnchors[i * 2] = anchors[m * 2];
usedAnchors[i * 2 + 1] = anchors[m * 2 + 1];
}
cv::Mat biasData_mat = cv::Mat(1, numAnchors * 2, CV_32F, &usedAnchors[0]).clone();
@@ -835,6 +837,7 @@ namespace cv {
tensor_shape[0] = 0;
for (size_t k = 0; k < layers_vec.size(); ++k) {
layers_vec[k] = layers_vec[k] >= 0 ? layers_vec[k] : (layers_vec[k] + layers_counter);
CV_Assert(layers_vec[k] >= 0 && static_cast<size_t>(layers_vec[k]) < net->out_channels_vec.size());
tensor_shape[0] += net->out_channels_vec[layers_vec[k]];
}