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 |
)
|