82a8953e5cc49d98826f9a621c7bc8f13d8314b7,examples/FasterRCNN/modeling/model_mrcnn.py,,maskrcnn_loss,#Any#Any#Any#,16

Before Change


        fg_labels: //fg, in 1~//class, int64
        fg_target_masks: //fgxhxw, float32
    
    num_fg = tf.size(fg_labels, out_type=tf.int64)
    indices = tf.stack([tf.range(num_fg), fg_labels - 1], axis=1)  // //fgx2
    mask_logits = tf.gather_nd(mask_logits, indices)  // //fgxhxw
    mask_probs = tf.sigmoid(mask_logits)

After Change


        fg_labels: //fg, in 1~//class, int64
        fg_target_masks: //fgxhxw, float32
    
    mask_logits = tf.batch_gather(mask_logits, tf.reshape(fg_labels, [-1, 1]) - 1)
    mask_logits = tf.squeeze(mask_logits, axis=1)
    mask_probs = tf.sigmoid(mask_logits)
Italian Trulli
In pattern: SUPERPATTERN

Frequency: 3

Non-data size: 3

Instances


Project Name: tensorpack/tensorpack
Commit Name: 82a8953e5cc49d98826f9a621c7bc8f13d8314b7
Time: 2019-08-26
Author: ppwwyyxx@users.noreply.github.com
File Name: examples/FasterRCNN/modeling/model_mrcnn.py
Class Name:
Method Name: maskrcnn_loss


Project Name: jhjacobsen/pytorch-i-revnet
Commit Name: 4b3ca2794fe54a4b71ab2bdaea57cfd01c2f2849
Time: 2020-05-06
Author: simon.pezold@web.de
File Name: models/model_utils.py
Class Name: psi
Method Name: forward


Project Name: uber/pyro
Commit Name: 6e899f41f3437058c79197084b35891f4e3f073f
Time: 2020-05-08
Author: info@stefanwebb.me
File Name: pyro/distributions/transforms/generalized_channel_permute.py
Class Name: GeneralizedChannelPermute
Method Name: _inverse