Function get_segment_indices

deeplabcut/utils/make_labeled_video.py:58–71

(bodyparts2connect, all_bpts)

def get_segment_indices(bodyparts2connect, all_bpts):
    bpts2connect = []
    for bpt1, bpt2 in bodyparts2connect:
        if bpt1 in all_bpts and bpt2 in all_bpts:
            bpts2connect.extend(
                zip(
                    *(
                        np.flatnonzero(all_bpts == bpt1),
                        np.flatnonzero(all_bpts == bpt2),
                    ),
                    strict=False,
                )
            )
    return bpts2connect


def CreateVideo(