Python niqeの例

コード例 #1

ファイル: test_metrics.py プロジェクト: Yshuo-Li/mmediting-test

def test_calculate_niqe():
    img = mmcv.imread('tests/data/gt/baboon.png')

    result = niqe(img[:, :, 0], crop_border=0, input_order='HW')
    np.testing.assert_almost_equal(result, 6.15902, decimal=5)
    result = niqe(img, crop_border=0, input_order='HWC', convert_to='y')
    np.testing.assert_almost_equal(result, 5.85182, decimal=5)
    result = niqe(img, crop_border=0, input_order='HWC', convert_to='gray')
    np.testing.assert_almost_equal(result, 5.89766, decimal=5)
    result = niqe(
        img.transpose(2, 0, 1),
        crop_border=0,
        input_order='CHW',
        convert_to='y')
    np.testing.assert_almost_equal(result, 5.85182, decimal=5)
    result = niqe(
        img.transpose(2, 0, 1),
        crop_border=0,
        input_order='CHW',
        convert_to='gray')
    np.testing.assert_almost_equal(result, 5.89766, decimal=5)

    result = niqe(img[:, :, 0], crop_border=6, input_order='HW')
    np.testing.assert_almost_equal(result, 6.31046, decimal=5)
    result = niqe(img, crop_border=6, input_order='HWC', convert_to='y')
    np.testing.assert_almost_equal(result, 6.14435, decimal=5)
    result = niqe(
        img.transpose(2, 0, 1),
        crop_border=6,
        input_order='CHW',
        convert_to='y')
    np.testing.assert_almost_equal(result, 6.14435, decimal=5)

コード例 #2

ファイル: test_metrics.py プロジェクト: open-mmlab/mmediting

def test_calculate_niqe():
    img = mmcv.imread('tests/data/gt/baboon.png')

    result = niqe(img[:, :, 0], crop_border=0, input_order='HW')
    np.testing.assert_almost_equal(result, 5.62525, decimal=5)
    result = niqe(img, crop_border=0, input_order='HWC', convert_to='y')
    np.testing.assert_almost_equal(result, 5.72957, decimal=5)
    result = niqe(img, crop_border=0, input_order='HWC', convert_to='gray')
    np.testing.assert_almost_equal(result, 5.73154, decimal=5)
    result = niqe(img.transpose(2, 0, 1),
                  crop_border=0,
                  input_order='CHW',
                  convert_to='y')
    np.testing.assert_almost_equal(result, 5.72957, decimal=5)
    result = niqe(img.transpose(2, 0, 1),
                  crop_border=0,
                  input_order='CHW',
                  convert_to='gray')
    np.testing.assert_almost_equal(result, 5.73154, decimal=5)

    result = niqe(img[:, :, 0], crop_border=6, input_order='HW')
    np.testing.assert_almost_equal(result, 5.82981, decimal=5)
    result = niqe(img, crop_border=6, input_order='HWC', convert_to='y')
    np.testing.assert_almost_equal(result, 6.10074, decimal=5)
    result = niqe(img.transpose(2, 0, 1),
                  crop_border=6,
                  input_order='CHW',
                  convert_to='y')
    np.testing.assert_almost_equal(result, 6.10074, decimal=5)