-
Notifications
You must be signed in to change notification settings - Fork 9
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Support more NWB training data types #100
Comments
11 tasks
6 tasks
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
As written in #86, some NWB data types are not supported yet in sleap-io, including
TrainingFrame
,TrainingFrames
,SourceVideo
, andPoseTraining
. #24 shows the data model equivalences between SLEAP and NWB, and I am expanding support for NWB data by allowing sleap-io to save and load it. I have written a functionconvert_nwb
innwb.py
that initializes new SLEAP objects when an NWB object is used as an argument.https://github.com/keyaloding/io_fork/blob/70a34ede294d3c783d1758c55c1dd9752e29f1e7/sleap_io/io/nwb.py#L37-L72
I am planning to write a new function that calls the
read_nwb
insleap_io/io/nwb.py
and returns the attributes of theLabels
object that is returned. It will return eitherLabels.labeled_frames
,Labels.videos
, orhttps://github.com/talmolab/sleap-io/blob/5a301106effd8cbcc6f23c9385e3c274b6b9262f/sleap_io/io/nwb.py#L40-L48 These attributes will correspond to the
TrainingFrame,
TrainingFrames,
SourceVideos, and
PoseTraining` data types. I will add new tests that run inference and training on .nwb dataThe text was updated successfully, but these errors were encountered: