Closed
Description
Keras losses reduce over one axis by default - currently IoU is reducing over two. This will make it impossible for us to mask out background boxes. To get around this, we should only reduce over one axis instead of two in IoU loss.
This means that after the IoU is computed you should have [batch_size, num_boxes] returned from the loss