13de22d42ecbc51ea19bdd571262959eb29804d9,frcnn/roi_pool.py,ROIPoolingLayer,_roi_crop,#ROIPoolingLayer#Any#Any#,39

Before Change

        bboxes = self._get_bboxes(rois, pretrained)
        // TODO: Why?!!?
        batch_ids = tf.squeeze(tf.slice(rois, [0, 0], [-1, 1], name="batch_id"), [1])

        crops = tf.image.crop_and_resize(
            pretrained, bboxes, tf.to_int32(batch_ids),
            [self._pooled_width * 2, self._pooled_height * 2], name="crops"

After Change

        // TODO: Why?!!?
        // batch_ids = tf.squeeze(tf.slice(roi_proposals, [0, 0], [-1, 1], name="batch_id"), [1])
        bboxes_shape = tf.shape(bboxes)
        batch_ids = tf.zeros((bboxes_shape[0], ), dtype=tf.int32)
        crops = tf.image.crop_and_resize(
            pretrained, bboxes, batch_ids,
            [self._pooled_width * 2, self._pooled_height * 2], name="crops"
        )

Instances