Skip to content

Commit

Permalink
Changed incorrect bbox format in loss_bbox docstring (facebookresearc…
Browse files Browse the repository at this point in the history
…h#90)

(center_x, center_y, h, w) changed to (center_x, center_y, w, h)
  • Loading branch information
hafiz703 authored Jun 15, 2020
1 parent 17f2ffa commit 1fcfc65
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion models/detr.py
Original file line number Diff line number Diff line change
Expand Up @@ -143,7 +143,7 @@ def loss_cardinality(self, outputs, targets, indices, num_boxes):
def loss_boxes(self, outputs, targets, indices, num_boxes):
"""Compute the losses related to the bounding boxes, the L1 regression loss and the GIoU loss
targets dicts must contain the key "boxes" containing a tensor of dim [nb_target_boxes, 4]
The target boxes are expected in format (center_x, center_y, h, w), normalized by the image size.
The target boxes are expected in format (center_x, center_y, w, h), normalized by the image size.
"""
assert 'pred_boxes' in outputs
idx = self._get_src_permutation_idx(indices)
Expand Down

0 comments on commit 1fcfc65

Please sign in to comment.