Features/region proposal network losses refactor

broadinstitute / keras-rcnn

Keras package for region-based convolutional neural networks (RCNNs)

Other

555 stars 222 forks source link

Closed hgaiser closed 7 years ago

hgaiser commented 7 years ago

This PR refactors the RPN/RCNN models. In addition it moves gather_nd and matmul to keras_rcnn.backend and enables rpn_regression_loss.

0x00b1 commented 7 years ago

Thanks, @hgaiser! I like this.

I might go back and refactor the RPN model so it's usable like a Keras layer.