CharlesShang / FastMaskRCNN

Mask RCNN in TensorFlow
Apache License 2.0
3.1k stars 1.1k forks source link

about the roi align backprop #116

Closed GriffithKQ closed 6 years ago

GriffithKQ commented 7 years ago

i can not find the backprop step in the all the packages, especially i examined the crop.py where seems like the roialign and find nothing. does any one find it ? if we do not have this layer , the result should not be coverage

animebing commented 7 years ago

@GriffithKQ, the roi align is implemented in the return statement of function crop_ in libs/layers/crop.py as below

    return  [tf.image.crop_and_resize(images, boxes, batch_inds,
                                         [pooled_height, pooled_width],
                                         method='bilinear',
                                         name='Crop')] + [boxes]

here I mean tf.image.crop_and_resize, so I think the back propagation is done by tensorflow