Deblur-NeRF复现出现的问题:raise ValueError( ValueError: win_size exceeds image extent. Either ensure ...

问题报错具体如下:

Traceback (most recent call last): File "run_nerf.py", line 653, in <module> train() File "run_nerf.py", line 623, in train test_ssim = compute_img_metric(rgbs, target_rgb_ldr, 'ssim') 
File "/root/autodl-tmp/Deblur-NeRF/metrics.py", line 81, in compute_img_metric value, ssimmap = photometric["ssim"]( File "/root/miniconda3/lib/python3.8/site-packages/skimage/metrics/_structural_similarity.py", line 178, in structural_similarity raise ValueError( ValueError: 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指标时出现的问题,同时出现在Deblur-NeRF和DP-NeRF复现的过程汇中,现将解决方法记录如下:

在metrics.py文件的81行左右将

value, ssimmap = photometric["ssim"](
                im1[i], im2[i], multichannel=True, full=True
                )

修改为:

value, ssimmap = photometric["ssim"](
                im1[i], im2[i], multichannel=True, full=True, channel_axis=-1, data_range=im1[i].max()-im1[i].min()
            )

  • 3
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值