Skip to content

GitLab

"...nn/git@developer.sourcefind.cn:OpenDAS/fairscale.git" did not exist on "15512d9e071c861f6cd9129eaa678d9d93ec371e"
Commit 7b165eb4 authored by Kaushik Shivakumar's avatar Kaushik Shivakumar
Browse files

fix tests

parent ba65cc7e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
research/object_detection/core/target_assigner.py
View file @ 7b165eb4
...@@ -142,6 +142,8 @@ class TargetAssigner(object): ...@@ -142,6 +142,8 @@ class TargetAssigner(object):
aware of groundtruth weights. Additionally, `cls_weights` and aware of groundtruth weights. Additionally, `cls_weights` and
`reg_weights` are calculated using groundtruth weights as an added `reg_weights` are calculated using groundtruth weights as an added
safety. safety.
class_predictions: a float tensor of shape [N, num_classes] containing class
predictions from the model, to be used in certain matchers.
Returns: Returns:
cls_targets: a float32 tensor with shape [num_anchors, d_1, d_2 ... d_k], cls_targets: a float32 tensor with shape [num_anchors, d_1, d_2 ... d_k],
......
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