|
|
@@ -1419,7 +1419,7 @@ class RoIHeads(nn.Module):
|
|
|
if gt_circles_tensor.shape[0] > 0:
|
|
|
print(f'start to compute circle_loss')
|
|
|
|
|
|
- loss_circle = compute_arc_loss(feature_logits, circle_proposals, gt_circles,
|
|
|
+ loss_circle = maskrcnn_loss(feature_logits, circle_proposals, gt_circles,
|
|
|
circle_pos_matched_idxs)
|
|
|
|
|
|
# loss_circle_extra = compute_circle_extra_losses(feature_logits, circle_proposals, gt_circles,circle_pos_matched_idxs)
|
admin