Issues with Image Outputs on Other Datasets

[zjc24]

Great work. When I test on other datasets, the images show the following situation. Could you please explain why this happens?

[ywyii123]

You might need to check if the data processing code is correct or if the pre-trained model was loaded properly.