mmdetection/tests/test_models/test_losses/test_l2_loss.py

# Copyright (c) OpenMMLab. All rights reserved.
from unittest import TestCase

import torch

from mmdet.models import L2Loss


class TestL2Loss(TestCase):

    def test_l2_loss(self):
        pred = torch.Tensor([[1, 1, 0, 0, 0, 0, 1]])
        target = torch.Tensor([[1, 1, 0, 0, 0, 0, 0]])

        loss = L2Loss(
            neg_pos_ub=2,
            pos_margin=0,
            neg_margin=0.1,
            hard_mining=True,
            loss_weight=1.0)
        assert torch.allclose(loss(pred, target), torch.tensor(0.1350))