AIDGE IN SPACE
Banc d'essai de compression d'image apprise pour l'embarqué
Chargement...
Recherche...
Aucune correspondance
Référence de l'espace de nommage aidge_in_space.metrics

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).

Description détaillée

Mesures de qualité (PSNR, MS-SSIM) et écart de débit de Bjøntegaard (BD-rate).

Documentation des fonctions

◆ 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.

Paramètres
rate_refDébits de la courbe de référence.
quality_refQualités (PSNR, MS-SSIM en dB...) de la référence.
rate_testDébits de la courbe testée.
quality_testQualités de la courbe testée.
Renvoie
Écart en % (négatif = le codec testé économise du débit), ou NaN s'il y a moins de 4 points par courbe ou pas de plage de qualité commune.

◆ ms_ssim()

aidge_in_space.metrics.ms_ssim ( a,
b )

MS-SSIM à 5 échelles (Wang et al.

2003), fenêtre gaussienne 11x11 de sigma 1,5.

Paramètres
aImage de référence (uint8).
bImage reconstruite, même forme.
Renvoie
MS-SSIM dans [0, 1], moyenné sur les canaux ; NaN si un côté est inférieur à MS_SSIM_MIN_SIDE.

◆ ms_ssim_db()

aidge_in_space.metrics.ms_ssim_db ( v)

MS-SSIM en décibels : -10 log10(1 - MS-SSIM), plus lisible près de 1.

Paramètres
vMS-SSIM (NaN accepté).
Renvoie
Valeur en dB, ou NaN.

◆ psnr()

aidge_in_space.metrics.psnr ( a,
b )

Peak Signal-to-Noise Ratio entre deux images 8 bits.

Paramètres
aImage de référence (uint8).
bImage reconstruite, même forme.
Renvoie
PSNR en dB (infini si les images sont identiques).

Documentation des variables

◆ MS_SSIM_WEIGHTS

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).