|
AIDGE IN SPACE
Banc d'essai de compression d'image apprise pour l'embarqué
|
Transformée d'analyse de Ballé 2017 : conv 9x9/4, 5x5/2, 5x5/2, chacune suivie d'une GDN. Plus de détails...
Fonctions membres publiques | |
| __init__ (self, in_ch=1, N=128, gdn_last=True) | |
| Crée les couches (poids non entraînés). | |
| forward (self, x) | |
| Image 1 x in_ch x H x W -> latent 1 x N x H/16 x W/16. | |
Attributs publics | |
| conv1 = nn.Conv2d(in_ch, N, 9, stride=4, padding=4) | |
| Conv 9x9 de pas 4. | |
| gdn1 = GDN(N, symmetric=True) | |
| GDN après conv1. | |
| conv2 = nn.Conv2d(N, N, 5, stride=2, padding=2) | |
| Conv 5x5 de pas 2. | |
| gdn2 = GDN(N, symmetric=True) | |
| GDN après conv2. | |
| conv3 = nn.Conv2d(N, N, 5, stride=2, padding=2) | |
| Conv 5x5 de pas 2. | |
| gdn3 = GDN(N, symmetric=True) if gdn_last else nn.Identity() | |
| GDN après conv3 (identité si gdn_last est faux). | |
Transformée d'analyse de Ballé 2017 : conv 9x9/4, 5x5/2, 5x5/2, chacune suivie d'une GDN.
| aidge_in_space.nn.Analysis2017.__init__ | ( | self, | |
| in_ch = 1, | |||
| N = 128, | |||
| gdn_last = True ) |
Crée les couches (poids non entraînés).
| in_ch | Canaux de l'image (1 en gris, 3 en RGB). |
| N | Canaux des couches et du latent. |
| gdn_last | False pour supprimer la dernière GDN. |