测试GridDehazeNet代码
skimage版本为0.21.0,太新了,会报错module 'skimage.measure' has no attribute 'compare_ssim'
改成
from skimage.metrics import structural_similarity as compare_ssim
又报错
alueError: win_size exceeds image extent. Either ensure that your images are at least 7x7; or pass win_size explicitly in the function call, with an odd value less than or equal to the smaller side of your images. If your images are multichannel (with color channels), set channel_axis to the axis number corresponding to the channels.
于是
源代码
ssim_list = [compare_ssim(dehaze_list_np[ind], gt_list_np[ind], data_range=1, multichannel=True) for ind in range(len(dehaze_list))] 改成 ssim_list = [compare_ssim(dehaze_list_np[ind], gt_list_np[ind], data_range=1, multichannel=True, channel_axis=2) for ind in range(len(dehaze_list))]
即加了一个channel_axis=2
来源于这位大佬
然后跑通了
Ok