update ocr_rec.cpp

ca7be744 · Double_V · GitHub · 6335b0d8 · ca7be744
Unverified Commit ca7be744 authored Feb 24, 2021 by Double_V Committed by GitHub Feb 24, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

deploy/cpp_infer/src/ocr_rec.cpp deploy/cpp_infer/src/ocr_rec.cpp +1 -1

No files found.
--- a/deploy/cpp_infer/src/ocr_rec.cpp
+++ b/deploy/cpp_infer/src/ocr_rec.cpp
@@ -76,7 +76,7 @@ void CRNNRecognizer::Run(std::vector<std::vector<std::vector<int>>> boxes,
          float(*std::max_element(&predict_batch[n * predict_shape[2]],
                                  &predict_batch[(n + 1) * predict_shape[2]]));

-      if (argmax_idx > 0 && (!(i >= 0 && argmax_idx == last_index))) {
+      if (argmax_idx > 0 && (!(n > 0 && argmax_idx == last_index))) {
        score += max_value;
        count += 1;
        str_res.push_back(label_list_[argmax_idx]);