AIDGE IN SPACE
Banc d'essai de compression d'image apprise pour l'embarqué
Chargement...
Recherche...
Aucune correspondance
Référence de la classe aidge_in_space.nn.Analysis2017

Transformée d'analyse de Ballé 2017 : conv 9x9/4, 5x5/2, 5x5/2, chacune suivie d'une GDN. Plus de détails...

Graphe d'héritage de aidge_in_space.nn.Analysis2017:
[légende]

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

Description détaillée

Transformée d'analyse de Ballé 2017 : conv 9x9/4, 5x5/2, 5x5/2, chacune suivie d'une GDN.

Documentation des constructeurs et destructeur

◆ __init__()

aidge_in_space.nn.Analysis2017.__init__ ( self,
in_ch = 1,
N = 128,
gdn_last = True )

Crée les couches (poids non entraînés).

Paramètres
in_chCanaux de l'image (1 en gris, 3 en RGB).
NCanaux des couches et du latent.
gdn_lastFalse pour supprimer la dernière GDN.

La documentation de cette classe a été générée à partir du fichier suivant :
  • nn.py