IIGROUP / MANIQA

[CVPRW 2022] MANIQA: Multi-dimension Attention Network for No-Reference Image Quality Assessment
Apache License 2.0
307 stars 36 forks source link

Get score for individual images #18

Closed Ellyuca closed 2 years ago

Ellyuca commented 2 years ago

Hi. Thanks for this interesting work. I am interested in using the inference.py script to get the individual score of each image in a folder? What parameters do I need to change in the configuration? I see there is this:

    # config file
    config = Config({
        # dataset path
        "db_name": "PIPAL",
        "test_dis_path": "/mnt/data_16TB/ysd21/IQA/NTIRE2022_NR_Valid_Dis/",

        # optimization
        "batch_size": 10,
        "num_avg_val": 1,
        "crop_size": 224,

        # device
        "num_workers": 8,

        # load & save checkpoint
        "valid": "./output/valid",
        "valid_path": "./output/valid/inference_valid",
        "model_path": "./output/models/model_maniqa/epoch1"
    })

Do I just need to change the batch_size from 10 to 1? Thank you.

TianheWu commented 2 years ago

yes.

Ellyuca commented 2 years ago

Thanks.