2de47a11683658e4a5fb70b8e98e8c99983605ec,test/test_metrics.py,TestMeanIoU,test_four_classes_2d_perfect,#TestMeanIoU#,36

Before Change

        mean_iou = tgm.metrics.mean_iou(predicted, actual, num_classes)
        assert mean_iou.shape == (1, num_classes)
        assert pytest.approx(mean_iou[..., 0].item(), 1.00)
        assert pytest.approx(mean_iou[..., 1].item(), 1.00)
        assert pytest.approx(mean_iou[..., 2].item(), 1.00)
        assert pytest.approx(mean_iou[..., 3].item(), 1.00)

    def test_four_classes_2d_one_class_no_predicted(self):

After Change

              [2, 2, 3, 3]]])

        mean_iou = tgm.metrics.mean_iou(predicted, actual, num_classes)
        mean_iou_real = torch.tensor(
            [[1.0, 1.0, 1.0, 1.0]], dtype=torch.float32)
        assert mean_iou.shape == (batch_size, num_classes)
        assert utils.check_equal_torch(mean_iou, mean_iou_real)

    def test_four_classes_2d_one_class_no_predicted(self):

Instances