Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2024-11-21 19:07:46 +01:00
Code Issues Projects Releases Wiki Activity

only resize when needed

Browse Source
This commit is contained in:
blakeblackshear 2019-03-17 20:12:31 -05:00
parent 8bae05cfe2
commit 862aa2d3f0
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
frigate/object_detection.py
View File

@ -24,7 +24,8 @@ def ReadLabelFile(file_path):
# do the actual object detection # do the actual object detection
def tf_detect_objects(cropped_frame, engine, labels, region_size, region_x_offset, region_y_offset, debug): def tf_detect_objects(cropped_frame, engine, labels, region_size, region_x_offset, region_y_offset, debug):
# Resize to 300x300 # Resize to 300x300 if needed
if cropped_frame.shape != (300, 300, 3):
cropped_frame = cv2.resize(cropped_frame, dsize=(300, 300), interpolation=cv2.INTER_LINEAR) cropped_frame = cv2.resize(cropped_frame, dsize=(300, 300), interpolation=cv2.INTER_LINEAR)
# Expand dimensions since the model expects images to have shape: [1, None, None, 3] # Expand dimensions since the model expects images to have shape: [1, None, None, 3]
image_np_expanded = np.expand_dims(cropped_frame, axis=0) image_np_expanded = np.expand_dims(cropped_frame, axis=0)