|
AIDGE IN SPACE
Banc d'essai de compression d'image apprise pour l'embarqué
|
Transformée de synthèse de Ballé 2017, symétrique de Analysis2017 (IGDN + conv transposées). Plus de détails...
Fonctions membres publiques | |
| __init__ (self, in_ch=1, N=128, gdn_last=True) | |
| Mêmes paramètres que Analysis2017. | |
| forward (self, y) | |
| Latent 1 x N x h x w -> image 1 x in_ch x 16h x 16w. | |
Attributs publics | |
| igdn1 = GDN(N, inverse=True, symmetric=True) if gdn_last else nn.Identity() | |
| IGDN d'entrée (identité si gdn_last est faux). | |
| deconv1 = nn.ConvTranspose2d(N, N, 5, stride=2, padding=2, output_padding=1) | |
| Conv transposée 5x5 de pas 2. | |
| igdn2 = GDN(N, inverse=True, symmetric=True) | |
| IGDN après deconv1. | |
| deconv2 = nn.ConvTranspose2d(N, N, 5, stride=2, padding=2, output_padding=1) | |
| Conv transposée 5x5 de pas 2. | |
| igdn3 = GDN(N, inverse=True, symmetric=True) | |
| IGDN après deconv2. | |
| deconv3 = nn.ConvTranspose2d(N, in_ch, 9, stride=4, padding=4, output_padding=3) | |
| Conv transposée 9x9 de pas 4, vers les canaux de l'image. | |
Transformée de synthèse de Ballé 2017, symétrique de Analysis2017 (IGDN + conv transposées).