Store motion data as a percent of total area

This commit is contained in:
Nicolas Mowen 2024-03-14 09:13:24 -06:00
parent 45a318dfed
commit fe38f450df
2 changed files with 20 additions and 29 deletions

View File

@ -368,30 +368,17 @@ def motion_activity():
camera_list = cameras.split(",") camera_list = cameras.split(",")
clauses.append((Recordings.camera << camera_list)) clauses.append((Recordings.camera << camera_list))
all_recordings: list[Recordings] = ( data: list[Recordings] = (
Recordings.select( Recordings.select(
Recordings.start_time, Recordings.start_time,
Recordings.duration,
Recordings.objects,
Recordings.motion, Recordings.motion,
) )
.where(reduce(operator.and_, clauses)) .where(reduce(operator.and_, clauses))
.order_by(Recordings.start_time.asc()) .order_by(Recordings.start_time.asc())
.dicts()
.iterator() .iterator()
) )
# format is: { timestamp: segment_start_ts, motion: [0-100], audio: [0 - -100] }
# periods where active objects / audio was detected will cause motion to be scaled down
data: list[dict[str, float]] = []
for rec in all_recordings:
data.append(
{
"start_time": rec.start_time,
"motion": rec.motion if rec.objects == 0 else 0,
}
)
# get scale in seconds # get scale in seconds
scale = request.args.get("scale", type=int, default=30) scale = request.args.get("scale", type=int, default=30)
@ -403,17 +390,7 @@ def motion_activity():
df.set_index(["start_time"], inplace=True) df.set_index(["start_time"], inplace=True)
# normalize data # normalize data
df = df.resample(f"{scale}S").sum().fillna(0.0) df = df.resample(f"{scale}S").mean().fillna(0.0)
mean = df["motion"].mean()
std = df["motion"].std()
df["motion"] = (df["motion"] - mean) / std
outliers = df.quantile(0.999)["motion"]
df[df > outliers] = outliers
df["motion"] = (
(df["motion"] - df["motion"].min())
/ (df["motion"].max() - df["motion"].min())
* 100
)
# change types for output # change types for output
df.index = df.index.astype(int) // (10**9) df.index = df.index.astype(int) // (10**9)

View File

@ -71,6 +71,13 @@ class RecordingMaintainer(threading.Thread):
self.audio_recordings_info: dict[str, list] = defaultdict(list) self.audio_recordings_info: dict[str, list] = defaultdict(list)
self.end_time_cache: dict[str, Tuple[datetime.datetime, float]] = {} self.end_time_cache: dict[str, Tuple[datetime.datetime, float]] = {}
self.camera_frame_area = dict[str, int]
for camera in self.config.cameras.values():
self.camera_frame_area[camera.name] = (
camera.detect.width * camera.detect.height
)
async def move_files(self) -> None: async def move_files(self) -> None:
cache_files = [ cache_files = [
d d
@ -289,8 +296,9 @@ class RecordingMaintainer(threading.Thread):
def segment_stats( def segment_stats(
self, camera: str, start_time: datetime.datetime, end_time: datetime.datetime self, camera: str, start_time: datetime.datetime, end_time: datetime.datetime
) -> SegmentInfo: ) -> SegmentInfo:
video_frame_count = 0
active_count = 0 active_count = 0
motion_count = 0 total_motion_area = 0
for frame in self.object_recordings_info[camera]: for frame in self.object_recordings_info[camera]:
# frame is after end time of segment # frame is after end time of segment
if frame[0] > end_time.timestamp(): if frame[0] > end_time.timestamp():
@ -299,6 +307,7 @@ class RecordingMaintainer(threading.Thread):
if frame[0] < start_time.timestamp(): if frame[0] < start_time.timestamp():
continue continue
video_frame_count += 1
active_count += len( active_count += len(
[ [
o o
@ -307,7 +316,12 @@ class RecordingMaintainer(threading.Thread):
] ]
) )
motion_count += sum([area(box) for box in frame[2]]) total_motion_area += sum([area(box) for box in frame[2]])
normalized_motion_area = int(
(total_motion_area / (self.camera_frame_area[camera] * video_frame_count))
* 100
)
audio_values = [] audio_values = []
for frame in self.audio_recordings_info[camera]: for frame in self.audio_recordings_info[camera]:
@ -327,7 +341,7 @@ class RecordingMaintainer(threading.Thread):
average_dBFS = 0 if not audio_values else np.average(audio_values) average_dBFS = 0 if not audio_values else np.average(audio_values)
return SegmentInfo(motion_count, active_count, round(average_dBFS)) return SegmentInfo(normalized_motion_area, active_count, round(average_dBFS))
async def move_segment( async def move_segment(
self, self,