Skip to content

Commit

Permalink
updating todos
Browse files Browse the repository at this point in the history
  • Loading branch information
SkepticRaven committed Sep 6, 2023
1 parent abaa5a6 commit 82141bd
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions sleap_io/io/jabs.py
Original file line number Diff line number Diff line change
Expand Up @@ -117,7 +117,6 @@ def read_labels(labels_path: str, skeleton: Optional[Skeleton] = JABS_DEFAULT_SK
frames.append(frame_label)
return Labels(frames)

Check warning on line 118 in sleap_io/io/jabs.py

View check run for this annotation

Codecov / codecov/patch

sleap_io/io/jabs.py#L109-L118

Added lines #L109 - L118 were not covered by tests


def prediction_to_instance(data: Union[np.ndarray[np.uint16], np.ndarray[np.float32]], confidence: np.ndarray[np.float32], skeleton: Skeleton, track: Track = None) -> Instance:
"""Create an `Instance` from prediction data.
Expand Down Expand Up @@ -160,8 +159,6 @@ def get_max_ids_in_video(labels: List[Labels], key: str = 'Mouse') -> int:

def convert_labels(all_labels: Labels, video: str) -> dict:
"""Convert a `Labels` object into JABS-formatted annotations.
TODO: Currently assumes all data is mouse
TODO: Identity is an unsafe str -> cast. Convert to factorize op
TODO: See ignored fields in `read_labels`
Args:
Expand Down Expand Up @@ -264,6 +261,7 @@ def write_jabs_v2(data: dict, filename: str):
def write_jabs_v3(data: dict, filename: str):
""" Write JABS pose file v3 data to file.
Writes multi-mouse pose data.
TODO: v3 requires continuous tracklets (eg no gaps) IDs need to be incremented for this field
Args:
data: Dictionary of JABS data generated from convert_labels
Expand Down

0 comments on commit 82141bd

Please sign in to comment.