"official/nlp/bert/model_training_utils.py" did not exist on "533d1e6b26f7fe3e4e355ec1a1aeb3277a059c8a"
Commit e0b082ed authored by Kaushik Shivakumar's avatar Kaushik Shivakumar
Browse files

fix documentation for box coder

parent 4f7965fb
......@@ -37,16 +37,10 @@ EPSILON = 1e-8
class DETRBoxCoder(box_coder.BoxCoder):
"""Faster RCNN box coder."""
"""DETR box coder."""
def __init__(self):
"""Constructor for DETRBoxCoder.
Args:
scale_factors: List of 4 positive scalars to scale ty, tx, th and tw.
If set to None, does not perform scaling. For Faster RCNN,
the open-source implementation recommends using [10.0, 10.0, 5.0, 5.0].
"""
"""Constructor for DETRBoxCoder."""
pass
@property
......@@ -58,10 +52,10 @@ class DETRBoxCoder(box_coder.BoxCoder):
Args:
boxes: BoxList holding N boxes to be encoded.
anchors: BoxList of anchors.
anchors: BoxList of anchors, ignored for DETR.
Returns:
a tensor representing N anchor-encoded boxes of the format
a tensor representing N encoded boxes of the format
[ty, tx, th, tw].
"""
# Convert anchors to the center coordinate representation.
......@@ -72,8 +66,8 @@ class DETRBoxCoder(box_coder.BoxCoder):
"""Decode relative codes to boxes.
Args:
rel_codes: a tensor representing N anchor-encoded boxes.
anchors: BoxList of anchors.
rel_codes: a tensor representing N encoded boxes.
anchors: BoxList of anchors, ignored for DETR.
Returns:
boxes: BoxList holding N bounding boxes.
......
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