questions about get_seg_single

[cizhenshi]

HI~I am reading source code of SOLO and I am not sure what strides are in the following code and what role has he taken? I'd appreciate it if you could give me a hint

    # process.
    inds = (cate_preds > cfg.score_thr)
    # category scores.
    cate_scores = cate_preds[inds]
    if len(cate_scores) == 0:
        return None
    # category labels.
    inds = inds.nonzero()
    cate_labels = inds[:, 1]

    # strides.
    size_trans = cate_labels.new_tensor(self.seg_num_grids).pow(2).cumsum(0)
    strides = cate_scores.new_ones(size_trans[-1])
    n_stage = len(self.seg_num_grids)
    strides[:size_trans[0]] *= self.strides[0]
    for ind_ in range(1, n_stage):
        strides[size_trans[ind_ - 1]:size_trans[ind_]] *= self.strides[ind_]
    strides = strides[inds[:, 0]]

    # masks.
    seg_preds = seg_preds[inds[:, 0]]
    seg_masks = seg_preds > cfg.mask_thr
    sum_masks = seg_masks.sum((1, 2)).float()

    # filter.
    keep = sum_masks > strides
    if keep.sum() == 0:
        return None`

[rivercn]

we are the same question