"torchvision/datapoints/_bounding_box.py" did not exist on "b8af91ab78d9e95ac377e059c7c47249128d5bdb"
test_niqe.py 399 Bytes
Newer Older
mashun1's avatar
mashun1 committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
import cv2
import warnings

from basicsr.metrics import calculate_niqe


def main():
    img_path = 'tests/data/baboon.png'
    img = cv2.imread(img_path)
    with warnings.catch_warnings():
        warnings.simplefilter('ignore', category=RuntimeWarning)
        niqe_result = calculate_niqe(img, 0, input_order='HWC', convert_to='y')
    print(niqe_result)


if __name__ == '__main__':
    main()