Skip to content

IoU Loss should only reduce over 1 axis by default. #762

Closed
@LukeWood

Description

@LukeWood

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

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions