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

Add area labels to bounding boxes (#47)

Browse Source 
* Add object size to the bounding box

Remove script from Dockerfile

Fix framerate command

Move default value for framerate

update dockerfile

dockerfile changes

Add person_area label to surrounding box


Update dockerfile


ffmpeg config bug


Add `person_area` label to `best_person` frame


Resolve debug view showing area label for non-persons


Add object size to the bounding box


Add object size to the bounding box

* Move object area outside of conditional to work with all object types
This commit is contained in:
Kyle Niewiada
2019-07-03 07:14:39 -04:00
committed by Blake Blackshear
parent 8fee8f86a2
commit c653567cc1
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
frigate/video.py
View File

@@ -234,8 +234,10 @@ class Camera:
return
for obj in objects:
# Store object area to use in bounding box labels
obj['area'] = (obj['xmax']-obj['xmin'])*(obj['ymax']-obj['ymin'])
if obj['name'] == 'person':
person_area = (obj['xmax']-obj['xmin'])*(obj['ymax']-obj['ymin'])
# find the matching region
region = None
for r in self.regions:
@@ -250,7 +252,7 @@ class Camera:
# if the min person area is larger than the
# detected person, don't add it to detected objects
if region and region['min_person_area'] > person_area:
if region and region['min_person_area'] > obj['area']:
continue
# compute the coordinates of the person and make sure
@@ -279,7 +281,7 @@ class Camera:
# draw the bounding boxes on the screen
for obj in detected_objects:
label = "{}: {}%".format(obj['name'],int(obj['score']*100))
label = "{}: {}% {}".format(obj['name'],int(obj['score']*100),int(obj['area']))
draw_box_with_label(frame, obj['xmin'], obj['ymin'], obj['xmax'], obj['ymax'], label)
for region in self.regions: