search

broadinstitute / keras-rcnn

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

Region proposal network (RPN) classification loss #11

Closed 0x00b1 closed 7 years ago

0x00b1 commented 7 years ago

It’s implanted here:

https://github.com/broadinstitute/keras-rcnn/blob/master/keras_rcnn/losses/proposals.py#L4-L16

but it needs documentation and unit tests.