def _get_data_slice_sparse(self, seq_idx, key, start_frame, end_frame): key_prefix = key[:key.index("[")] sparse_info = key[key.index("[") + 1:key.index("]")].split(":") assert len(sparse_info) == 4 assert tuple(sparse_info[0:3]) == ("sparse", "coo", "2") s0 = self.get_data(seq_idx, "%s[sparse:coo:2:0]" % key_prefix) assert s0 is not None from NativeOp import sparse_splice_offset_numpy s0_start = sparse_splice_offset_numpy(s0, start_frame) s0_end = sparse_splice_offset_numpy(s0, end_frame) if sparse_info[-1] == "0": return s0[s0_start:s0_end] - start_frame else: data = self.get_data(seq_idx, key) return data[s0_start:s0_end]
def _get_data_slice_sparse(self, seq_idx, key, start_frame, end_frame): key_prefix = key[:key.index("[")] sparse_info = key[key.index("[") + 1:key.index("]")].split(":") assert len(sparse_info) == 4 assert tuple(sparse_info[0:3]) == ("sparse", "coo", "2") s0 = self.get_data(seq_idx, "%s[sparse:coo:2:0]" % key_prefix) assert s0 is not None from NativeOp import sparse_splice_offset_numpy s0_start = sparse_splice_offset_numpy(s0, start_frame) s0_end = sparse_splice_offset_numpy(s0, end_frame) if sparse_info[-1] == "0": return s0[s0_start:s0_end] - start_frame else: data = self.get_data(seq_idx, key) return data[s0_start:s0_end]