Commit 7695c2ff authored by dmMaze's avatar dmMaze
Browse files

apply dilate to imgtrans_pipeline's mask

parent 2abf351f

ballontranslator/ui/dl_manager.py

+4 −0
Original line number Diff line number Diff line
@@ -256,6 +256,7 @@ class ImgtransThread(QThread): 
        self.inpaint_thread = inpaint_thread

        self.job = None

        self.imgtrans_proj: ProjImgTrans = None

        self.mask_postprocess = None



    @property

    def textdetector(self) -> TextDetectorBase:
@@ -293,6 +294,9 @@ class ImgtransThread(QThread): 
            img = self.imgtrans_proj.read_img(imgname)



            mask, blk_list = self.textdetector.detect(img)

            if self.mask_postprocess is not None:

                mask = self.mask_postprocess(mask)

                

            self.imgtrans_proj.save_mask(imgname, mask)



            self.detect_counter += 1

ballontranslator/ui/mainwindow.py

+1 −0
Original line number Diff line number Diff line
@@ -182,6 +182,7 @@ class MainWindow(QMainWindow): 
        dl_manager.imgtrans_pipeline_finished.connect(self.on_imgtrans_pipeline_finished)

        dl_manager.page_trans_finished.connect(self.on_pagtrans_finished)

        dl_manager.progress_msgbox.showed.connect(self.on_imgtrans_progressbox_showed)

        dl_manager.imgtrans_thread.mask_postprocess = self.drawingPanel.rectPanel.post_process_mask



        self.leftBar.run_imgtrans.connect(self.on_run_imgtrans)

        self.bottomBar.ocrcheck_statechanged.connect(dl_manager.setOCRMode)