90b78b1379b3423aa3b4b5d0de7313f88bda23ab,chainercv/utils/mask/mask_to_bbox.py,,mask_to_bbox,#Any#,5

Before Change


    bbox = []
    for msk in mask:
        where = np.argwhere(msk)
        if len(where) > 0:
            y_min, x_min = where.min(0)
            y_max, x_max = where.max(0) + 1
        else:
            y_min, x_min, y_max, x_max = 0, 0, 0, 0
        bbox.append((y_min, x_min, y_max, x_max))
    if len(bbox) == 0:
        return xp.empty((0, 4), dtype=np.float32)
    return xp.array(bbox, dtype=np.float32)

After Change



    instance_index, ys, xs = xp.nonzero(mask)
    bbox = xp.zeros((R, 4), dtype=np.float32)
    for i in range(R):
        ys_i = ys[instance_index == i]
        xs_i = xs[instance_index == i]
        if len(ys_i) == 0:
            continue
        y_min = ys_i.min()
        x_min = xs_i.min()
        y_max = ys_i.max() + 1
        x_max = xs_i.max() + 1
        bbox[i] = xp.array([y_min, x_min, y_max, x_max], dtype=np.float32)
    return bbox
Italian Trulli
In pattern: SUPERPATTERN

Frequency: 3

Non-data size: 8

Instances


Project Name: chainer/chainercv
Commit Name: 90b78b1379b3423aa3b4b5d0de7313f88bda23ab
Time: 2019-02-17
Author: yuyuniitani@gmail.com
File Name: chainercv/utils/mask/mask_to_bbox.py
Class Name:
Method Name: mask_to_bbox


Project Name: pytorch/fairseq
Commit Name: cf1c64a5f7cffe7a362cc772752f8f226fb04f22
Time: 2018-06-15
Author: myleott@fb.com
File Name: train.py
Class Name:
Method Name: validate