"vscode:/vscode.git/clone" did not exist on "d55e5cc9039b8107cace43bc7c28ea62b266c656"
Unverified Commit 0e9d5840 authored by Srihari Humbarwadi's avatar Srihari Humbarwadi Committed by GitHub
Browse files

Fixed typos in losses.py

parent df7e66d2
...@@ -449,7 +449,7 @@ class RetinanetBoxLoss(object): ...@@ -449,7 +449,7 @@ class RetinanetBoxLoss(object):
num_positives: number of positive examples in the minibatch. num_positives: number of positive examples in the minibatch.
Returns: Returns:
an integar tensor representing total box regression loss. an integer tensor representing total box regression loss.
""" """
# Sums all positives in a batch for normalization and avoids zero # Sums all positives in a batch for normalization and avoids zero
# num_positives_sum, which would lead to inf loss during training # num_positives_sum, which would lead to inf loss during training
...@@ -457,7 +457,6 @@ class RetinanetBoxLoss(object): ...@@ -457,7 +457,6 @@ class RetinanetBoxLoss(object):
box_losses = [] box_losses = []
for level in box_outputs.keys(): for level in box_outputs.keys():
# Onehot encoding for classification labels.
box_targets_l = labels[level] box_targets_l = labels[level]
box_losses.append( box_losses.append(
self.box_loss(box_outputs[level], box_targets_l, num_positives_sum)) self.box_loss(box_outputs[level], box_targets_l, num_positives_sum))
......
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