tf.image.ssim_multiscale
| Просмотреть исходный код на GitHub |
Вычисляет MS-SSIM между img1 и img2.
tf.image.ssim_multiscale(
img1, img2, max_val, power_factors=_MSSSIM_WEIGHTS, filter_size=11,
filter_sigma=1.5, k1=0.01, k2=0.03
)
Эта функция предполагает, что img1 и img2 являются пакетами изображений, т. е. последние три измерения — [высота, ширина, каналы].
Примечание: Истинный SSIM определен только для градации серого. Эта функция не выполняет преобразование цветового пространства. (Если вход уже YUV, она вычисляет среднее значение YUV SSIM).
Оригинальная статья: Wang, Zhou, Eero P. Simoncelli, and Alan C. Bovik. «Мультимасштабная структурная схожесть для оценки качества изображений». Сигналы, системы и компьютеры, 2004.
| Аргументы | |
|---|---|
img1 | Первый пакет изображений. |
img2 | Второй пакет изображений. Должен иметь тот же ранг, что и img1. |
max_val | Диапазон динамики изображений (т. е. разность между максимальным и минимальным допустимыми значениями). |
power_factors | Итерируемый список весов для каждого масштаба. Количество используемых масштабов равно длине списка. Индекс 0 — это вес без масштабирования, и каждый увеличивающийся масштаб соответствует уменьшению размера изображения в 2 раза. По умолчанию используется (0,0448, 0,2856, 0,3001, 0,2363, 0,1333), что соответствует значениям, полученным в оригинальной статье. |
filter_size | Значение по умолчанию 11 (размер гауссова фильтра). |
filter_sigma | Значение по умолчанию 1,5 (ширина гауссова фильтра). |
k1 | Значение по умолчанию 0,01 |
k2 | Значение по умолчанию 0,03 (SSIM менее чувствителен к K2 при меньших значениях, поэтому лучше брать значения в диапазоне 0 < K2 < 0,4). |
| Возвращаемое значение | |
|---|---|
| Тензор, содержащий значение MS-SSIM для каждого изображения в пакете. Значения находятся в диапазоне [0, 1]. Возвращает тензор с формой: broadcast(img1.shape[:-3], img2.shape[:-3]). |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/image/ssim_multiscale