broadinstitute / keras-rcnn

Keras package for region-based convolutional neural networks (RCNNs)
Other
553 stars 223 forks source link

Shape of y_pred and y_true #31

Closed jhung0 closed 7 years ago

jhung0 commented 7 years ago

Why is the shape of y_pred and y_true (a, b, c, d) (e.g. see https://github.com/broadinstitute/keras-rcnn/blob/master/keras_rcnn/losses/rpn.py#L57)? I think it'd make more sense to have shape (m, n). https://github.com/mitmul/chainer-faster-rcnn also supports that.