juanpablomesa commited on
Commit
1dfe4c5
·
1 Parent(s): 374a5b3

Instead of OCR every frame, doit every 8 video frames

Browse files
Files changed (1) hide show
  1. handler.py +6 -1
handler.py CHANGED
@@ -195,10 +195,15 @@ class EndpointHandler:
195
  # Extract text from each frame using EasyOCR
196
  self.logger.info("Extracting text from frames.")
197
  text_extraction_start_time = timeit.default_timer()
198
- frame_texts = [self.reader.readtext(frame, detail=0) for frame in frames]
 
 
 
 
199
  texts_set = set()
200
  for text_list in frame_texts:
201
  [texts_set.add(text) for text in text_list]
 
202
  video_metadata["extracted_text"] = json.dumps(
203
  texts_set, default=self.set_default
204
  )
 
195
  # Extract text from each frame using EasyOCR
196
  self.logger.info("Extracting text from frames.")
197
  text_extraction_start_time = timeit.default_timer()
198
+ # frame_texts = [self.reader.readtext(frame, detail=0) for frame in frames]
199
+ frame_texts = [
200
+ self.reader.readtext(frames[i], detail=0)
201
+ for i in range(0, len(frames), 8)
202
+ ]
203
  texts_set = set()
204
  for text_list in frame_texts:
205
  [texts_set.add(text) for text in text_list]
206
+
207
  video_metadata["extracted_text"] = json.dumps(
208
  texts_set, default=self.set_default
209
  )