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

working odroid build, still needs hwaccel

Browse Source
This commit is contained in:
blakeblackshear
2019-05-27 10:17:57 -05:00
committed by Blake Blackshear
parent 56b9c754f5
commit 9f8278ea8f
3 changed files with 64 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
frigate/objects.py
View File

@@ -2,7 +2,6 @@ import time
import datetime
import threading
import cv2
from object_detection.utils import visualization_utils as vis_util
class ObjectCleaner(threading.Thread):
def __init__(self, objects_parsed, detected_objects):
@@ -82,15 +81,10 @@ class BestPersonFrame(threading.Thread):
best_frame = recent_frames[self.best_person['frame_time']]
best_frame = cv2.cvtColor(best_frame, cv2.COLOR_BGR2RGB)
# draw the bounding box on the frame
vis_util.draw_bounding_box_on_image_array(best_frame,
self.best_person['ymin'],
self.best_person['xmin'],
self.best_person['ymax'],
self.best_person['xmax'],
color='red',
thickness=2,
display_str_list=["{}: {}%".format(self.best_person['name'],int(self.best_person['score']*100))],
use_normalized_coordinates=False)
color = (255,0,0)
cv2.rectangle(best_frame, (self.best_person['xmin'], self.best_person['ymin']),
(self.best_person['xmax'], self.best_person['ymax']),
color, 2)
# convert back to BGR
self.best_frame = cv2.cvtColor(best_frame, cv2.COLOR_RGB2BGR)