search

broadinstitute / keras-rcnn

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

Detection layer argmax and NMS #117

Closed jhung0 closed 6 years ago

jhung0 commented 6 years ago

After applying deltas to proposals, only take coordinates corresponding to the class with the highest score. Then do NMS of the boxes.