Closed Ellyuca closed 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.
yes.
Thanks.
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:
Do I just need to change the batch_size from 10 to 1? Thank you.