fix problem of yolo_box

PaddlePaddle · Mar 2, 2022 · d16f0fa · d16f0fa · paddle-bot-old · Mar 2, 2022
1 parent 95dbb89
commit d16f0fa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/fluid/operators/detection/yolo_box_op.cc b/paddle/fluid/operators/detection/yolo_box_op.cc
@@ -103,7 +103,7 @@ class YoloBoxOp : public framework::OperatorWithKernel {
                           class_num));
 
     int box_num;
-    if (dim_x[2] > 0 && dim_x[3] > 0 || ctx->IsRuntime()) {
+    if ((dim_x[2] > 0 && dim_x[3] > 0) || ctx->IsRuntime()) {
       box_num = dim_x[2] * dim_x[3] * anchor_num;
     } else {
       box_num = -1;