def arange_as_seq()

in mapillary_sls/datasets/msls.py [0:0]

• 12 lines of code
• 8 McCabe index (conditional complexity)

    def arange_as_seq(self, data, path, seq_length):

        seqInfo = pd.read_csv(join(path, 'seq_info.csv'), index_col = 0)

        seq_keys, seq_idxs = [], []
        for idx in data.index:

            # edge cases.
            if idx < (seq_length//2) or idx >= (len(seqInfo) - seq_length//2): continue

            # find surrounding frames in sequence
            seq_idx = np.arange(-seq_length//2, seq_length//2) + 1 + idx
            seq = seqInfo.iloc[seq_idx]

            # the sequence must have the same sequence key and must have consecutive frames
            if len(np.unique(seq['sequence_key'])) == 1 and (seq['frame_number'].diff()[1:] == 1).all():
                seq_key = ','.join([join(path, 'images', key + '.jpg') for key in seq['key']])

                seq_keys.append(seq_key)
                seq_idxs.append(seq_idx)

        return seq_keys, np.asarray(seq_idxs)