Merge branch 'feat_1' of 10.214.199.191:xuanweiace/accessibility_movie_2 into feat_1

detect_with_ocr.py

@@ -65,6 +65,8 @@ def evaluate_position(video_path: str, start_time: float) -> Tuple[float, float]
         for j in range(10):#每个阶段取10帧
             for k in range(120):#隔120帧取一帧
                 _,frame = video.read()
+            if (j+1)*120+frame_index>frame_num:
+                break
             _,frame = video.read()
             res = ocr.ocr(frame, cls=True)
             for result_item in res:
@@ -86,6 +88,8 @@ def evaluate_position(video_path: str, start_time: float) -> Tuple[float, float]
                                 position_map[(y1,y3)]=1
     print(sorted( position_map.items(), key = lambda kv:(kv[1], kv[0])))
     y1,y2 = sorted( position_map.items(), key = lambda kv:(kv[1], kv[0]))[-1][0]    
+    y1 = y1/video.get(cv2.CAP_PROP_FRAME_HEIGHT)
+    y2 = y2/video.get(cv2.CAP_PROP_FRAME_HEIGHT)
     return y1,y2
 
 ocr_positions = []