Commit 80b5a66c authored by dmMaze's avatar dmMaze
Browse files

fix postprocess of new ocr

parent b1c7f7d8

dl/ocr/mit48px_ctc.py

+5 −4
Original line number Diff line number Diff line
@@ -442,6 +442,7 @@ class OCR48pxCTC: 
            for i, single_line in enumerate(texts) :

                if not single_line :

                    continue

                textblk = textblk_lst[textblk_lst_indices[i+chunck_idx]]

                cur_texts = []

                total_fr = AvgMeter()

                total_fg = AvgMeter()
@@ -467,12 +468,12 @@ class OCR48pxCTC: 
                if prob < 0.3 :

                    continue

                textblk.text.append(''.join(cur_texts))

                textblk.fg_r += int(total_fr())

                textblk.fg_b += int(total_fr())

                textblk.fg_g += int(total_fg())

                textblk.fg_b += int(total_fb())

                textblk.bg_r += int(total_br())

                textblk.fg_r += int(total_fb())

                textblk.bg_b += int(total_br())

                textblk.bg_g += int(total_bg())

                textblk.bg_b += int(total_bb())

                textblk.bg_r += int(total_bb())

            chunck_idx += N