Skip to content

multibox loss #27

Open
Open
@XDUSPONGE

Description

Thanks for your code
in the paper

  • the cls loss is normalized by the number of positive and negative anchors
  • the reg term is normalized by the number of positive anchors.
    so may be
    image
    it should be changed
M = num_pos.data.sum().float() + num_neg.data.sum().float()
loss_c /= N

also in the paper L =\lambda * L_{cls} + L_{reg}
in the code

loss_l, loss_c = criterion(out, targets)
loss = loss_l + loss_c

may be the two place should be modified

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions