chris/frigate
1
0
Fork 0
forked from Github/frigate
Code Pull Requests Packages Activity

experimental: running ffmpeg directly and capturing raw frames

Browse Source
This commit is contained in:
blakeblackshear
2019-05-27 12:41:52 -05:00
committed by Blake Blackshear
parent 9f8278ea8f
commit 2b51dc3e5b
3 changed files with 41 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
frigate/object_detection.py
View File

@@ -90,12 +90,12 @@ class FramePrepper(threading.Thread):
frame_time = self.frame_time.value
# convert to RGB
cropped_frame_rgb = cv2.cvtColor(cropped_frame, cv2.COLOR_BGR2RGB)
#cropped_frame_rgb = cv2.cvtColor(cropped_frame, cv2.COLOR_BGR2RGB)
# Resize to 300x300 if needed
if cropped_frame_rgb.shape != (300, 300, 3):
cropped_frame_rgb = cv2.resize(cropped_frame_rgb, dsize=(300, 300), interpolation=cv2.INTER_LINEAR)
if cropped_frame.shape != (300, 300, 3):
cropped_frame = cv2.resize(cropped_frame, dsize=(300, 300), interpolation=cv2.INTER_LINEAR)
# Expand dimensions since the model expects images to have shape: [1, 300, 300, 3]
frame_expanded = np.expand_dims(cropped_frame_rgb, axis=0)
frame_expanded = np.expand_dims(cropped_frame, axis=0)
# add the frame to the queue
if not self.prepped_frame_queue.full():