Fix CUDA/CPU device mismatch in quality/eval.py

[politecat314]

When running quality/eval.py with device="cuda", the following error occurred on line 43:

pred_score = net(input_data).data.cpu().numpy().squeeze()

RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same or input should be a MKLDNN tensor and weight is a dense tensor

This error occurred because the model was on cuda but the input_data was on cpu.

Changes made to resolve the issue:

1. Modified read_img function to accept a device parameter and move the input tensor to the specified device.
2. Updated the device selection in the main script to use torch.device() with a check for CUDA availability.
3. Ensured that the input data is moved to the same device as the model before forward pass.

These changes ensure that both the model and input data are on the same device, resolving the CUDA/CPU mismatch error.