TensorFlow函數(shù)：tf.image.ssim_multiscale

tf.image.ssim_multiscale函數(shù)

tf.image.ssim_multiscale(
    img1,
    img2,
    max_val,
    power_factors=_MSSSIM_WEIGHTS
)

定義在：tensorflow/python/ops/image_ops_impl.py.

計算img1和img2之間的MS-SSIM.

該函數(shù)假設(shè)img1和img2是圖像批處理,即最后三個維度是[height, width, channels].

注意：真正的SSIM僅在灰度上定義.此功能不執(zhí)行任何色彩空間轉(zhuǎn)換.(如果輸入已經(jīng)是YUV,那么它將計算平均YUV SSIM.)

參數(shù)：

• img1：第一批圖像.
• img2：第二批圖像,必須與img1具有相同的秩.
• max_val：圖像的動態(tài)范圍(即最大允許值與最小允許值之間的差值).
• power_factors：每個量表的權(quán)重都是可迭代的,所用的比例數(shù)是列表的長度,索引0是未縮放的分辨率的權(quán)重,每個增加的縮放對應(yīng)于被2采樣的圖像,默認(rèn)為(0.0448,0.2856,0.3001,0.2363,0.1333).

返回：

包含批處理中每個圖像的MS-SSIM值的張量,值在[0,1]范圍內(nèi),返回帶形狀：broadcast(img1.shape [： - 3],img2.shape [： - 3])的張量.