Commit
·
1dfe4c5
1
Parent(s):
374a5b3
Instead of OCR every frame, doit every 8 video frames
Browse files- 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 |
)
|