search

broadinstitute / keras-rcnn

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

fix rcnn regression loss #123

Closed jhung0 closed 6 years ago

jhung0 commented 6 years ago

2 main edits:

  1. the labels matrix that was supposed to 0 out the coordinates corresponding to other classes (non ground truth classes) was only applied to outputs and not abs(outputs-target)
  2. the mean was found by dividing the sum of losses by the number of ones in the labels matrix, but that has 1 for each coordinate rather than each object, so the labels matrix was switched out for labels target