Référence de la classe aidge_in_space.codecs.balle2017.Balle2017Codec

Modèles Ballé 2017 d'une même architecture (et d'un même run, pour une série), un niveau par λ. Plus de détails...

Graphe d'héritage de aidge_in_space.codecs.balle2017.Balle2017Codec:

Fonctions membres publiques
	__init__ (self, in_ch, N, gdn_last, entries, series=None)
	Crée le codec d'une architecture, ou d'une série de la plateforme d'entraînement.
	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)
	Charge les poids du niveau dans le modèle.
	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
	N
	Architecture commune aux checkpoints : canaux de l'image, canaux des couches, GDN finale.
	gdn_last = in_ch, N, gdn_last
	Architecture commune aux checkpoints : canaux de l'image, canaux des couches, GDN finale.
	series = series
	Nom du run de la série, ou None pour les checkpoints du notebook.
	entries
	Checkpoints triés par λ croissant (donc par débit croissant), puis par numéro de job pour une série.

Membres hérités additionnels
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.
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

Modèles Ballé 2017 d'une même architecture (et d'un même run, pour une série), un niveau par λ.

Documentation des constructeurs et destructeur

__init__()

aidge_in_space.codecs.balle2017.Balle2017Codec.__init__	(		self,
			in_ch,
			N,
			gdn_last,
			entries,
			series = None )

Crée le codec d'une architecture, ou d'une série de la plateforme d'entraînement.

Paramètres

in_ch	Canaux de l'image (1 = gris, 3 = RGB).
N	Canaux des couches.
gdn_last	Présence de la dernière GDN.
entries	Checkpoints, dictionnaires avec les clés path, lmbda et step ; pour une série aussi job_id, name, author et complete.
series	Nom du run d'une série (checkpoints écrits par la plateforme), ou None pour les checkpoints du notebook, qui n'ont pas de run.

Documentation des fonctions membres

architecture()

aidge_in_space.codecs.balle2017.Balle2017Codec.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

level

Level du codec.

Renvoie

Description de l'architecture.

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

levels()

aidge_in_space.codecs.balle2017.Balle2017Codec.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 le chemin de son checkpoint.

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

load_weights()

aidge_in_space.codecs.balle2017.Balle2017Codec.load_weights	(		self,
			model,
			level )

Charge les poids du niveau dans le modèle.

Paramètres

model	Modèle renvoyé par make_model(level).
level	Level du codec.

Exceptions

Exception

si les poids sont introuvables ou incompatibles.

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

make_model()

aidge_in_space.codecs.balle2017.Balle2017Codec.make_model	(		self,
			level )

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

Paramètres

level

Level 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 :

• balle2017.py