fix bug

tools/infer/predict_rec.py

@@ -122,7 +122,7 @@ class TextRecognizer(object):
                     blank = probs.shape[1]
                     valid_ind = np.where(ind != (blank - 1))[0]
                     score = np.mean(probs[valid_ind, ind[valid_ind]])
-                    if not valid_ind:
+                    if len(valid_ind) == 0:
                         continue
                     # rec_res.append([preds_text, score])
                     rec_res[indices[beg_img_no + rno]] = [preds_text, score]

tools/infer_rec.py

@@ -99,7 +99,7 @@ def main():
             ind = np.argmax(probs, axis=1)
             blank = probs.shape[1]
             valid_ind = np.where(ind != (blank - 1))[0]
-            if not valid_ind:
+            if len(valid_ind) == 0:
                 continue
             score = np.mean(probs[valid_ind, ind[valid_ind]])
         elif loss_type == "attention":