search

broadinstitute / keras-rcnn

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

Detection layer #101

Closed jhung0 closed 6 years ago

jhung0 commented 6 years ago

Edits to detection layer: don't do NMS, just apply deltas and output detection coordinates and scores

Also add anchor output to AnchorTarget so that this layer can come after RPN in order to see RPN detections. This make AnchorTarget more similar to ProposalTarget as well.