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

Merge pull request #28467 from Sikandar1310291:fix/topk-layer-k-boundary-check-28445

Browse Source 
Fix: TopK layer K boundary check allows K == input_dim (#28445)
This commit is contained in:
Alexander Smorkalov
2026-01-27 14:15:37 +03:00
committed by GitHub
parent 774c7e01b3 d8f8267700
commit cf1177e87e
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/dnn/src/layers/topk_layer.cpp
+2 -2
View File
@@ -97,9 +97,9 @@ public:
// Normalize axis
int axis_normalized = normalize_axis(axis, input_shape.size());
// Check if K is in range (0, input_shape[axis])
// Check if K is in range (0, input_shape[axis]]
CV_CheckGT(K, 0, "TopK: K needs to be a positive integer");
CV_CheckLT(K, input_shape[axis_normalized], "TopK: K is out of range");
CV_CheckLE(K, input_shape[axis_normalized], "TopK: K is out of range");
// Assign output shape
auto output_shape = input_shape;