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.codecs.bmshj2018.Pretrained2018

Base des modèles 2018 pré-entraînés : 8 niveaux de qualité, poids téléchargés à la demande. Plus de détails...

Graphe d'héritage de aidge_in_space.codecs.bmshj2018.Pretrained2018:
[légende]

Fonctions membres publiques

 levels (self, mode)
 Niveaux proposés, du plus bas au plus haut débit.
 make_model (self, level)
 Architecture non entraînée du niveau (suffit au profilage, sans téléchargement).
 load_weights (self, model, level)
 Télécharge (une fois, empreinte vérifiée) puis charge les poids CompressAI.
 architecture (self, level)
 Texte court qui distingue deux architectures, par exemple « N=128, M=192 ».
Fonctions membres publiques hérités de aidge_in_space.codecs.base.LearnedCodec
 __init__ (self)
 Initialise le cache des modèles chargés.
 model (self, level)
 Modèle entraîné du niveau, chargé au premier appel puis gardé en cache (LRU).
 encoder (self, level)
 Transformée d'analyse g_a aux GDN gelées : ce qui part dans ONNX, Aidge et sur la carte.
 onboard_module (self, model)
 Tout ce qui s'exécute à bord pour produire le flux (hors codage arithmétique).
 run (self, img, level, mode, backend="torch", real_rate=True)
 Compresse puis décompresse une image.
 decode_latents (self, model, stream)
 Relit l'en-tête d'un flux de run() puis décode ses latents.
Fonctions membres publiques hérités de aidge_in_space.codecs.base.Codec
 available (self)
 Indique si le codec peut s'exécuter sur cette machine.
 __repr__ (self)
 Représentation courte : classe et identifiant.

Attributs publics statiques

 arch = None
 Classe d'architecture de aidge_in_space.nn, construite avec channels(qualité).
Attributs publics statiques hérités de aidge_in_space.codecs.base.LearnedCodec
int in_ch = 3
 Canaux d'entrée du modèle (3 pour les modèles RGB, qui reçoivent une image grise répliquée).
 coding = FactorizedCoding()
 Codage entropique des latents (aidge_in_space.coding.LatentCoding).
int max_cached_models = 4
 Nombre de modèles chargés gardés en mémoire (cache LRU).
Attributs publics statiques hérités de aidge_in_space.codecs.base.Codec
str key = ""
 Identifiant unique, utilisé par la ligne de commande et dans les CSV.
str label = ""
 Nom affiché.
str family = ""
 "Classique" ou "Appris" ; les codecs classiques sont listés en premier.
str description = ""
 Description courte affichée dans l'onglet « Codecs et aide ».
tuple modes = ("gris", "rgb")
 Modes d'image acceptés : "gris" et/ou "rgb".
tuple backends = ()
 Moteurs d'exécution possibles pour l'encodeur (voir aidge_in_space.backends) ; vide = sans objet.

Membres hérités additionnels

Attributs protégés hérités de aidge_in_space.codecs.base.LearnedCodec
 _cache = OrderedDict()
 Modèles chargés, indexés par Level.value, du moins au plus récemment utilisé.

Description détaillée

Base des modèles 2018 pré-entraînés : 8 niveaux de qualité, poids téléchargés à la demande.

Documentation des fonctions membres

◆ architecture()

aidge_in_space.codecs.bmshj2018.Pretrained2018.architecture ( self,
level )

Texte court qui distingue deux architectures, par exemple « N=128, M=192 ».

Deux niveaux de même architecture ont le même coût embarqué : l'onglet « Encodeur embarqué » ne profile qu'un niveau par architecture.

Paramètres
levelLevel du codec.
Renvoie
Description de l'architecture.

Réimplémentée à partir de aidge_in_space.codecs.base.LearnedCodec.

◆ levels()

aidge_in_space.codecs.bmshj2018.Pretrained2018.levels ( self,
mode )

Niveaux proposés, du plus bas au plus haut débit.

Paramètres
mode"gris" ou "rgb".
Renvoie
Liste de Level.

La valeur de chaque niveau est la qualité CompressAI (1 à 8).

Réimplémentée à partir de aidge_in_space.codecs.base.Codec.

◆ load_weights()

aidge_in_space.codecs.bmshj2018.Pretrained2018.load_weights ( self,
model,
level )

Télécharge (une fois, empreinte vérifiée) puis charge les poids CompressAI.

Paramètres
modelModèle renvoyé par make_model(level).
levelLevel du codec.
Exceptions
Exceptionsi le téléchargement échoue ou si le checkpoint est incompatible.

Réimplémentée à partir de aidge_in_space.codecs.base.LearnedCodec.

◆ make_model()

aidge_in_space.codecs.bmshj2018.Pretrained2018.make_model ( self,
level )

Architecture non entraînée du niveau (suffit au profilage, sans téléchargement).

Paramètres
levelLevel du codec.
Renvoie
torch.nn.Module respectant le contrat décrit dans la classe.

Réimplémentée à partir de aidge_in_space.codecs.base.LearnedCodec.


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