disabling back_prop wrt to bounding box coordinates.

broadinstitute / keras-rcnn

Keras package for region-based convolutional neural networks (RCNNs)

Other

555 stars 222 forks source link

disabling back_prop wrt to bounding box coordinates. #131

Open Abhijit-2592 opened 6 years ago

Abhijit-2592 commented 6 years ago

In the Google's paper under section 3.4 it is said that:

However we disable backpropagation with respect to bounding box coordinates as we have found this to be unstable during training.

This can now be implemented using keras.backend.stop_gradient() I guess. But I am not sure how to implement it. Any ideas? @0x00b1 @yhenon @hgaiser Thanks