aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/argaze/AreaOfInterest/AOIFeatures.py210
1 files changed, 113 insertions, 97 deletions
diff --git a/src/argaze/AreaOfInterest/AOIFeatures.py b/src/argaze/AreaOfInterest/AOIFeatures.py
index 4a109a8..548382c 100644
--- a/src/argaze/AreaOfInterest/AOIFeatures.py
+++ b/src/argaze/AreaOfInterest/AOIFeatures.py
@@ -46,7 +46,7 @@ class AreaOfInterest(numpy.ndarray):
return self.shape[1]
@property
- def size(self) -> int:
+ def points_number(self) -> int:
"""Number of points defining the area."""
return self.shape[0]
@@ -58,12 +58,29 @@ class AreaOfInterest(numpy.ndarray):
return self.shape[0] == 0
@property
+ def bounds(self) -> numpy.array:
+ """Get area's bounds."""
+
+ min_bounds = numpy.min(self, axis=0)
+ max_bounds = numpy.max(self, axis=0)
+
+ return numpy.array([min_bounds, max_bounds])
+
+ @property
def center(self) -> numpy.array:
"""Center of mass."""
return self.mean(axis=0)
@property
+ def size(self) -> numpy.array:
+ """Get scene size."""
+
+ min_bounds, max_bounds = self.bounds
+
+ return max_bounds - min_bounds
+
+ @property
def area(self) -> float:
"""Area of the polygon defined by aoi's points."""
@@ -75,14 +92,14 @@ class AreaOfInterest(numpy.ndarray):
!!! warning
Available for 2D AOI only."""
- assert(self.size > 1)
+ assert(self.points_number > 1)
assert(self.dimension == 2)
min_x, min_y = numpy.min(self, axis=0)
max_x, max_y = numpy.max(self, axis=0)
return numpy.array([(min_x, min_y), (max_x, min_y), (max_x, max_y), (min_x, max_y)])
-
+
def clockwise(self) -> AreaOfInterestType:
"""Get area points in clockwise order.
!!! warning
@@ -203,100 +220,6 @@ class AreaOfInterest(numpy.ndarray):
center_pixel = numpy.rint(self.center).astype(int)
cv2.circle(frame, center_pixel, 1, color, -1)
-AOIFrameType = TypeVar('AOIFrame', bound="AOIFrame")
-# Type definition for type annotation convenience
-
-class AOIFrame():
- """Define frame to draw into 2D AOI."""
-
- def __init__(self, aoi: AreaOfInterestType, size: tuple):
- """
- !!! warning
- Available for 2D AOI only."""
-
- assert(aoi.dimension == 2)
-
- self.__rX, self.__rY = size
-
- # Init coordinates
- self.__Sx = numpy.linspace(0., self.__rX/self.__rY, self.__rX)
- self.__Sy = numpy.linspace(0., 1., self.__rY)
-
- # Init heatmap
- self.heatmap_init()
-
- def point_spread(self, point: tuple, sigma: float):
- """Draw gaussian point spread into frame."""
-
- div = -2 * sigma**2
-
- x = point[0] / self.__rY # we use rY not rX !!!
- y = point[1] / self.__rY
-
- dX2 = (self.__Sx - x)**2
- dY2 = (self.__Sy - y)**2
-
- v_dX, v_dY = numpy.array(numpy.meshgrid(dX2, dY2)).reshape(2, -1)
-
- return numpy.exp((v_dX + v_dY) / div).reshape(self.__rY, self.__rX)
-
- def heatmap_init(self, buffer_size: int = 0):
- """Initialize heatmap matrix."""
-
- self.__point_spread_sum = numpy.zeros((self.__rY, self.__rX))
- self.__point_spread_buffer = []
- self.__point_spread_buffer_size = buffer_size
-
- def heatmap_update(self, point: tuple, sigma: float):
- """Update heatmap matrix.
-
- !!! danger
- Call heatmap_init() method before any update."""
-
- point_spread = self.point_spread(point, sigma)
-
- # Sum point spread
- self.__point_spread_sum += point_spread
-
- # If point spread buffering enabled
- if self.__point_spread_buffer_size > 0:
-
- self.__point_spread_buffer.append(point_spread)
-
- # Remove oldest point spread buffer frame
- if len(self.__point_spread_buffer) > self.__point_spread_buffer_size:
-
- self.__point_spread_sum -= self.__point_spread_buffer.pop(0)
-
- # Edit heatmap
- heatmap_gray = (255 * self.__point_spread_sum / numpy.max(self.__point_spread_sum)).astype(numpy.uint8)
- self.__heatmap_matrix = cv2.applyColorMap(heatmap_gray, cv2.COLORMAP_JET)
-
- @property
- def heatmap_buffer(self) -> int:
- """Get size of heatmap buffer."""
-
- return self.__point_spread_buffer_size
-
- @heatmap_buffer.setter
- def heatmap_buffer(self, size: int):
- """Set size of heatmap buffer (0 means no buffering)."""
-
- self.__point_spread_buffer = []
- self.__point_spread_buffer_size = size
-
- @property
- def heatmap(self):
- """Get heatmap matrix."""
-
- try:
-
- return self.__heatmap_matrix
-
- except AttributeError:
-
- return numpy.zeros((self.__rY, self.__rX, 3)).astype(numpy.uint8)
-
AOISceneType = TypeVar('AOIScene', bound="AOIScene")
# Type definition for type annotation convenience
@@ -446,3 +369,96 @@ class TimeStampedAOIScenes(DataStructures.TimeStampedBuffer):
super().__setitem__(ts, scene)
+AOIFrameType = TypeVar('AOIFrame', bound="AOIFrame")
+# Type definition for type annotation convenience
+
+class AOIFrame():
+ """Define frame to draw into 2D AOI."""
+
+ def __init__(self, aoi: AreaOfInterestType, size: tuple):
+ """
+ !!! warning
+ Available for 2D AOI only."""
+
+ assert(aoi.dimension == 2)
+
+ self.__rX, self.__rY = size
+
+ # Init coordinates
+ self.__Sx = numpy.linspace(0., self.__rX/self.__rY, self.__rX)
+ self.__Sy = numpy.linspace(0., 1., self.__rY)
+
+ # Init heatmap
+ self.heatmap_init()
+
+ def point_spread(self, point: tuple, sigma: float):
+ """Draw gaussian point spread into frame."""
+
+ div = -2 * sigma**2
+
+ x = point[0] / self.__rY # we use rY not rX !!!
+ y = point[1] / self.__rY
+
+ dX2 = (self.__Sx - x)**2
+ dY2 = (self.__Sy - y)**2
+
+ v_dX, v_dY = numpy.array(numpy.meshgrid(dX2, dY2)).reshape(2, -1)
+
+ return numpy.exp((v_dX + v_dY) / div).reshape(self.__rY, self.__rX)
+
+ def heatmap_init(self, buffer_size: int = 0):
+ """Initialize heatmap matrix."""
+
+ self.__point_spread_sum = numpy.zeros((self.__rY, self.__rX))
+ self.__point_spread_buffer = []
+ self.__point_spread_buffer_size = buffer_size
+
+ def heatmap_update(self, point: tuple, sigma: float):
+ """Update heatmap matrix.
+
+ !!! danger
+ Call heatmap_init() method before any update."""
+
+ point_spread = self.point_spread(point, sigma)
+
+ # Sum point spread
+ self.__point_spread_sum += point_spread
+
+ # If point spread buffering enabled
+ if self.__point_spread_buffer_size > 0:
+
+ self.__point_spread_buffer.append(point_spread)
+
+ # Remove oldest point spread buffer frame
+ if len(self.__point_spread_buffer) > self.__point_spread_buffer_size:
+
+ self.__point_spread_sum -= self.__point_spread_buffer.pop(0)
+
+ # Edit heatmap
+ heatmap_gray = (255 * self.__point_spread_sum / numpy.max(self.__point_spread_sum)).astype(numpy.uint8)
+ self.__heatmap_matrix = cv2.applyColorMap(heatmap_gray, cv2.COLORMAP_JET)
+
+ @property
+ def heatmap_buffer(self) -> int:
+ """Get size of heatmap buffer."""
+
+ return self.__point_spread_buffer_size
+
+ @heatmap_buffer.setter
+ def heatmap_buffer(self, size: int):
+ """Set size of heatmap buffer (0 means no buffering)."""
+
+ self.__point_spread_buffer = []
+ self.__point_spread_buffer_size = size
+
+ @property
+ def heatmap(self):
+ """Get heatmap matrix."""
+
+ try:
+
+ return self.__heatmap_matrix
+
+ except AttributeError:
+
+ return numpy.zeros((self.__rY, self.__rX, 3)).astype(numpy.uint8)