Unverified Commit b7aa30c2 authored by Kai Chen's avatar Kai Chen Committed by GitHub
Browse files

Merge pull request #279 from qianyizhang/master

fix docstring
parents f3a939fa 6c600c9e
...@@ -88,8 +88,8 @@ class MaxIoUAssigner(BaseAssigner): ...@@ -88,8 +88,8 @@ class MaxIoUAssigner(BaseAssigner):
"""Assign w.r.t. the overlaps of bboxes with gts. """Assign w.r.t. the overlaps of bboxes with gts.
Args: Args:
overlaps (Tensor): Overlaps between n bboxes and k gt_bboxes, overlaps (Tensor): Overlaps between k gt_bboxes and n bboxes,
shape(n, k). shape(k, n).
gt_labels (Tensor, optional): Labels of k gt_bboxes, shape (k, ). gt_labels (Tensor, optional): Labels of k gt_bboxes, shape (k, ).
Returns: Returns:
......
...@@ -16,7 +16,7 @@ def bbox_overlaps(bboxes1, bboxes2, mode='iou', is_aligned=False): ...@@ -16,7 +16,7 @@ def bbox_overlaps(bboxes1, bboxes2, mode='iou', is_aligned=False):
foreground). foreground).
Returns: Returns:
ious(Tensor): shape (n, k) if is_aligned == False else shape (n, 1) ious(Tensor): shape (m, n) if is_aligned == False else shape (m, 1)
""" """
assert mode in ['iou', 'iof'] assert mode in ['iou', 'iof']
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment