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

fix mask filtering

Browse Source
This commit is contained in:
Blake Blackshear 2020-01-09 06:50:53 -06:00
parent 02e1035826
commit d31ba69b1b
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
frigate/objects.py
View File

@ -114,8 +114,8 @@ class RegionRefiner(threading.Thread):
detected_objects = self.camera.detected_objects[frame_time].copy() detected_objects = self.camera.detected_objects[frame_time].copy()
# print(f"{frame_time} finished") # print(f"{frame_time} finished")
detected_object_groups = defaultdict(lambda: [])
# group by name # group by name
detected_object_groups = defaultdict(lambda: [])
for obj in detected_objects: for obj in detected_objects:
detected_object_groups[obj['name']].append(obj) detected_object_groups[obj['name']].append(obj)
@ -202,8 +202,8 @@ class RegionRefiner(threading.Thread):
# compute the coordinates of the object and make sure # compute the coordinates of the object and make sure
# the location isnt outside the bounds of the image (can happen from rounding) # the location isnt outside the bounds of the image (can happen from rounding)
y_location = min(int(obj['ymax']), len(self.camera.mask)-1) y_location = min(int(obj['box']['ymax']), len(self.camera.mask)-1)
x_location = min(int((obj['xmax']-obj['xmin'])/2.0)+obj['xmin'], len(self.camera.mask[0])-1) x_location = min(int((obj['box']['xmax']-obj['box']['xmin'])/2.0)+obj['box']['xmin'], len(self.camera.mask[0])-1)
# if the object is in a masked location, don't add it to detected objects # if the object is in a masked location, don't add it to detected objects
if self.camera.mask[y_location][x_location] == [0]: if self.camera.mask[y_location][x_location] == [0]: