修改已知2.0的bug

cpython666 · Dec 17, 2023 · 90d2c25 · 90d2c25
1 parent 3cb7741
commit 90d2c25
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/output/5996c51c176a374e448190cfe6f69837.txt b/output/5996c51c176a374e448190cfe6f69837.txt
diff --git a/output/【PyQtPySide界面美化】qt-material极简上手！.txt b/output/【PyQtPySide界面美化】qt-material极简上手！.txt
diff --git a/output/【开箱即用】手把手教学，PyInstaller打包程序一键搞定！.txt b/output/【开箱即用】手把手教学，PyInstaller打包程序一键搞定！.txt
diff --git a/视频字幕提取-OCR版-2.0.py b/视频字幕提取-OCR版-2.0.py
@@ -86,7 +86,7 @@ def get_video_zimu_imgs(self):
                     cv2.imencode(".png", new_frame)[1].tofile(image_save_path)
                     self.progress_zimu_signal.emit(frame_count // frame_interval,total_frames // frame_interval)
                     # break
-                    # # TODO
+                    # TODO
                 frame_count += 1
 
         cap.release()
@@ -96,13 +96,16 @@ def extract_zimu_from_file(self, file):
         zimu_width = self.image_width
         print(file)
         if data:
-            # print(data)
             ocrResult = data['ocrResult']
+            ocrResult=[i for i in ocrResult if i['location']['right'] and i['location']['left']]
+            # ocrResult=[i for i in ocrResult if i['location']['right'] and i['location']['left'] and i['location']['top'] and i['location']['bottom']]
             if ocrResult:
+                # 方案一：最宽字幕区间
+                ocrResult.sort(key=lambda x: -1 * (x['location']['right'] - x['location']['left']))
+                # 方案二：最高字幕区间
                 # ocrResult.sort(key=lambda x: (x['location']['top'] - x['location']['bottom']))
-                # ocrResult.sort(key=lambda x: -1 * (x['location']['right'] - x['location']['left']))
-                ocrResult=[i for i in ocrResult if i['location']['right'] and i['location']['left'] and i['location']['top'] and i['location']['bottom']]
-                ocrResult.sort(key=lambda x: -1 * (x['location']['right'] - x['location']['left'])*(x['location']['bottom'] - x['location']['top']))
+                # 方案三：最大面积
+                # ocrResult.sort(key=lambda x: -1 * (x['location']['right'] - x['location']['left'])*(x['location']['bottom'] - x['location']['top']))
                 center = ocrResult[0]['location']['left'] + ocrResult[0]['location']['right']
                 if abs(center - zimu_width) < 100:
                     print(ocrResult[0]['text'])