search

broadinstitute / keras-rcnn

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

Combine the loss layers #107

Open 0x00b1 opened 6 years ago

0x00b1 commented 6 years ago

keras-rcnn has two region proposal network (RPN) loss layers (RPNClassificationLoss and RPNRegressionLoss) and two region-based convolutional neural network (RCNN) layers (RCNNClassification and RCNNRegression). I think it makes sense to combine the two RPN layers into one RPN layer and the two RCNN layers into one RCNN layer in an effort to further simplify the keras-rcnn’s public API.

hgaiser commented 6 years ago

So there'd be one layer computing the RPN loss and one layer to compute the RCNN loss? I like that idea!