BasicSR

1
import cv2
2
import warnings
3

4
from basicsr.metrics import calculate_niqe
5

6

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

15

16
if __name__ == '__main__':
17
    main()
18