|
AIDGE IN SPACE
Banc d'essai de compression d'image apprise pour l'embarqué
|
Mesures de qualité (PSNR, MS-SSIM) et écart de débit de Bjøntegaard (BD-rate). Plus de détails...
Fonctions | |
| psnr (a, b) | |
| Peak Signal-to-Noise Ratio entre deux images 8 bits. | |
| _as_tensor (img) | |
| Image uint8 -> tenseur float64 1 x C x H x W. | |
| _ssim_and_cs (x, y, win, data_range=255.0) | |
| SSIM et terme de contraste-structure moyens par canal, fenêtre gaussienne séparable sans padding. | |
| ms_ssim (a, b) | |
| MS-SSIM à 5 échelles (Wang et al. | |
| ms_ssim_db (v) | |
| MS-SSIM en décibels : -10 log10(1 - MS-SSIM), plus lisible près de 1. | |
| bd_rate (rate_ref, quality_ref, rate_test, quality_test) | |
| Écart de débit moyen de Bjøntegaard, à qualité égale. | |
Variables | |
| MS_SSIM_WEIGHTS = torch.tensor([0.0448, 0.2856, 0.3001, 0.2363, 0.1333], dtype=torch.float64) | |
| Poids des 5 échelles de MS-SSIM (Wang et al. | |
| int | SSIM_WINDOW = 11 |
| Taille de la fenêtre gaussienne de SSIM. | |
| int | MS_SSIM_MIN_SIDE = SSIM_WINDOW * 2 ** (len(MS_SSIM_WEIGHTS) - 1) |
| Plus petit côté d'image pour MS-SSIM : après 4 sous-échantillonnages par 2 (arrondis vers le bas), l'échelle la plus grossière doit encore contenir la fenêtre (176 // 16 = 11). | |
Mesures de qualité (PSNR, MS-SSIM) et écart de débit de Bjøntegaard (BD-rate).
| aidge_in_space.metrics.bd_rate | ( | rate_ref, | |
| quality_ref, | |||
| rate_test, | |||
| quality_test ) |
Écart de débit moyen de Bjøntegaard, à qualité égale.
Interpolation cubique de log(débit) en fonction de la qualité, intégrée sur la plage de qualité commune aux deux courbes. Les points non finis ou de débit nul sont ignorés.
| rate_ref | Débits de la courbe de référence. |
| quality_ref | Qualités (PSNR, MS-SSIM en dB...) de la référence. |
| rate_test | Débits de la courbe testée. |
| quality_test | Qualités de la courbe testée. |
| aidge_in_space.metrics.ms_ssim | ( | a, | |
| b ) |
MS-SSIM à 5 échelles (Wang et al.
2003), fenêtre gaussienne 11x11 de sigma 1,5.
| a | Image de référence (uint8). |
| b | Image reconstruite, même forme. |
| aidge_in_space.metrics.ms_ssim_db | ( | v | ) |
MS-SSIM en décibels : -10 log10(1 - MS-SSIM), plus lisible près de 1.
| v | MS-SSIM (NaN accepté). |
| aidge_in_space.metrics.psnr | ( | a, | |
| b ) |
Peak Signal-to-Noise Ratio entre deux images 8 bits.
| a | Image de référence (uint8). |
| b | Image reconstruite, même forme. |
| aidge_in_space.metrics.MS_SSIM_WEIGHTS = torch.tensor([0.0448, 0.2856, 0.3001, 0.2363, 0.1333], dtype=torch.float64) |
Poids des 5 échelles de MS-SSIM (Wang et al.
2003).