import os
import cv2
import warnings
import argparse

from basicsr.metrics import calculate_niqe

parser = argparse.ArgumentParser()
parser.add_argument('--root_path', type=str, default='results')
args = parser.parse_args()

def main():
    for imgname in os.listdir(args.root_path):
        img_path = os.path.join(args.root_path, imgname)
        # 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()