0

我想用 R 做一些简单的图像比较(我不使用 python 的原因是工作流在 R 中)。我试图在 R 的包中搜索 ms-ssim 实现,但确实找到了除spatialcompare::msssim. 然而,正如我昨天在帖子中提到的,我发现这个函数的结果可能与我的输入不正确(可能与matrix转换有关raster?)。对于适当的 ms-ssim 代码还有更多建议吗?我实际上基于自我实现了一个,SpatialPack::SSIM因为一次又一次地对图像进行下采样似乎很容易,但不确定我是否正确编写它。我会把它作为答案。

4

1 回答 1

0

我的一个非常简单的 ms-ssim 的演示代码:

calcMSSSIM<-function(Mat1,Mat2){
  weight = c(0.0448, 0.2856, 0.3001, 0.2363, 0.1333)
  level<-5
  array<-c()
  ssim<-SpatialPack::SSIM(Mat1, Mat2, alpha=1, beta=1, gamma=1,
                    eps=c(0.01,0.03), L=max(tmp1,tmp2))$SSIM  
  array[1]<-ssim$comps["contrast"]*ssim$comps["structure"]
  result<-abs(array[1])^weight[1]
  for (i in 2:level) {
    tmp1 <- EBImage::resize(tmp1, w=dim(tmp1)[1]/2, filter="none")
    tmp2 <- EBImage::resize(tmp2, w=dim(tmp2)[1]/2, filter="none")
    ssim <- SpatialPack::SSIM(tmp1,tmp2,alpha=1, beta=1, gamma=1, 
                             eps=c(0.01,0.03), L=max(tmp1,tmp2))$SSIM
    array[i]<-ssim$comps["contrast"]*ssim$comps["structure"]
    result <- result*abs(array[i])^weight[i]
  }
  lum<-ssim$comps["luminance"]
  result<-result*lum^weight[i]
  return(list(array,result))
}

这是对的吗?

于 2021-02-25T09:16:50.543 回答