keras-rcnn icon indicating copy to clipboard operation
keras-rcnn copied to clipboard

Move RPN and RCNN losses into keras_rcnn.losses module

Open 0x00b1 opened this issue 6 years ago • 0 comments

The RPN and RCNN layers have methods for their respective losses (each has a classification_loss and a regression_loss). We should refactor the methods into functions and move them into the keras_rcnn.losses module to replicate the Keras interface.

In addition, they should, wherever necessary, be refactored to conform to the Keras loss API:

def name(y_true, y_pred):
  return …

and expose the useful Keras alias interface:

model.compile("RCNN", None)

0x00b1 avatar Jan 30 '18 18:01 0x00b1