SainsburyWellcomeCentre / aeon_mecha

Project Aeon's main library for interfacing with acquired data. Contains modules for raw data file io, data querying, data processing, data qc, database ingestion, and building computational data pipelines.
BSD 3-Clause "New" or "Revised" License
3 stars 5 forks source link

Support social0.3 data, update SLEAP tracking with the new Pose Reader (Bonsai-sleap 3.0), bugfix BlockDetection #378

Closed ttngu207 closed 2 months ago

ttngu207 commented 2 months ago
github-actions[bot] commented 2 months ago

Image description CodeRabbit

Walkthrough

The changes primarily focus on enhancing the logic of block analysis, tracking, and schema updates. The block analysis now has improved detection of block end times and timestamp handling. Tracking adjustments include better retrieval of body parts and classes, and pose data processing based on identities. Schema updates involve support for social_03 data, updated SLEAP tracking, and bugfixes in BlockDetection. New Pose classes have been added to both social_01.py and social_03.py files.

Changes

File Summary
aeon/dj_pipeline/analysis/block_analysis.py Improved logic for detecting block end times, handling timestamps, updating block_end based on data frames, and ensuring consistency. Modifications related to subject data processing and updating block duration.
aeon/dj_pipeline/tracking.py Adjustments in retrieving body parts and classes, and processing pose data based on identities. Modifications to handle identity likelihood values.
aeon/schema/schemas.py Added support for social_03 data, updated SLEAP tracking with the new Pose Reader (Bonsai-sleap 3.0), and bugfixed BlockDetection. Changes include importing social_03 schema, updating device configurations for social_01 and social_03, adding a new social_03 device configuration, and including social_03 in the exported variables.
aeon/schema/social_01.py, aeon/schema/social_03.py Both files have added a new Pose class inheriting from Stream with an __init__ method that initializes using _reader.Pose.

Uplevel your code reviews with CodeRabbit Pro ### CodeRabbit Pro If you like this project, please support us by purchasing the [Pro version](https://coderabbit.ai). The Pro version has advanced context, superior noise reduction and several proprietary improvements compared to the open source version. Moreover, CodeRabbit Pro is free for open source projects.