Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-01-21 00:06:44 +01:00
Code Issues Projects Releases Wiki Activity

allow specific objects to be listed for a zone (fixes #1123)

Browse Source
This commit is contained in:
Blake Blackshear 2021-07-07 07:31:42 -05:00
parent da2492413f
commit f4aa02cc19
3 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
docs/docs/configuration/cameras.md
View File

@ -93,6 +93,9 @@ zones:
# Required: List of x,y coordinates to define the polygon of the zone. # Required: List of x,y coordinates to define the polygon of the zone.
# NOTE: Coordinates can be generated at https://www.image-map.net/ # NOTE: Coordinates can be generated at https://www.image-map.net/
coordinates: 545,1077,747,939,788,805 coordinates: 545,1077,747,939,788,805
# Optional: List of objects that can trigger this zone (default: all tracked objects)
objects:
- person
# Optional: Zone level object filters. # Optional: Zone level object filters.
# NOTE: The global and camera filters are applied upstream. # NOTE: The global and camera filters are applied upstream.
filters: filters:
@ -312,6 +315,9 @@ cameras:
# Required: List of x,y coordinates to define the polygon of the zone. # Required: List of x,y coordinates to define the polygon of the zone.
# NOTE: Coordinates can be generated at https://www.image-map.net/ # NOTE: Coordinates can be generated at https://www.image-map.net/
coordinates: 545,1077,747,939,788,805 coordinates: 545,1077,747,939,788,805
# Optional: List of objects that can trigger this zone (default: all tracked objects)
objects:
- person
# Optional: Zone level object filters. # Optional: Zone level object filters.
# NOTE: The global and camera filters are applied upstream. # NOTE: The global and camera filters are applied upstream.
filters: filters:

4
frigate/config.py
View File

@ -185,6 +185,10 @@ class ZoneConfig(BaseModel):
coordinates: Union[str, List[str]] = Field( coordinates: Union[str, List[str]] = Field(
title="Coordinates polygon for the defined zone." title="Coordinates polygon for the defined zone."
) )
objects: List[str] = Field(
default_factory=list,
title="List of objects that can trigger the zone.",
)
_color: Optional[Tuple[int, int, int]] = PrivateAttr() _color: Optional[Tuple[int, int, int]] = PrivateAttr()
_contour: np.ndarray = PrivateAttr() _contour: np.ndarray = PrivateAttr()

3
frigate/object_processing.py
View File

@ -143,6 +143,9 @@ class TrackedObject:
bottom_center = (self.obj_data["centroid"][0], self.obj_data["box"][3]) bottom_center = (self.obj_data["centroid"][0], self.obj_data["box"][3])
# check each zone # check each zone
for name, zone in self.camera_config.zones.items(): for name, zone in self.camera_config.zones.items():
# if the zone is not for this object type, skip
if len(zone.objects) > 0 and not self.obj_data["label"] in zone.objects:
continue
contour = zone.contour contour = zone.contour
# check if the object is in the zone # check if the object is in the zone
if cv2.pointPolygonTest(contour, bottom_center, False) >= 0: if cv2.pointPolygonTest(contour, bottom_center, False) >= 0: