AIDGE IN SPACE
Banc d'essai de compression d'image apprise pour l'embarqué
Chargement...
Recherche...
Aucune correspondance
Référence de l'espace de nommage aidge_in_space.images

Jeux d'images et conversions entre tableaux uint8 et tenseurs. Plus de détails...

Fonctions

 download_kodak (dest=KODAK_DIR)
 Télécharge les 24 images Kodak, en sautant celles déjà présentes.
 list_images (folder)
 Images d'un dossier et de ses sous-dossiers, triées par chemin.
 datasets ()
 Jeux d'images disponibles : Kodak, puis chaque sous-dossier de data/ (sauf train/).
 convert (img, mode)
 Convertit une image PIL ou un tableau dans le mode demandé.
 load_image (path, mode)
 Charge une image depuis un fichier, convertie avec convert().
 to_tensor (img, channels=None)
 Tableau uint8 -> tenseur 1 x C x H x W dans [0, 1].
 from_tensor (t, mode)
 Tenseur 1 x C x H x W dans [0, 1] -> tableau uint8 dans le mode demandé.

Variables

dict MODE_LABELS = {"gris": "Niveaux de gris", "rgb": "Couleur (RGB)"}
 Modes d'image : identifiant -> libellé affiché.
dict IMAGE_EXTS = {".png", ".jpg", ".jpeg", ".tif", ".tiff", ".bmp"}
 Extensions reconnues comme images.

Description détaillée

Jeux d'images et conversions entre tableaux uint8 et tenseurs.

Documentation des fonctions

◆ convert()

aidge_in_space.images.convert ( img,
mode )

Convertit une image PIL ou un tableau dans le mode demandé.

Paramètres
imgImage PIL ou tableau numpy.
mode"gris" ou "rgb".
Renvoie
Tableau uint8, H x W (gris) ou H x W x 3 (rgb).

◆ datasets()

aidge_in_space.images.datasets ( )

Jeux d'images disponibles : Kodak, puis chaque sous-dossier de data/ (sauf train/).

Renvoie
Dictionnaire nom -> liste de chemins (seuls les jeux non vides sont listés).

◆ download_kodak()

aidge_in_space.images.download_kodak ( dest = KODAK_DIR)

Télécharge les 24 images Kodak, en sautant celles déjà présentes.

Un échec (hors ligne) est signalé sans lever d'exception.

Paramètres
destDossier de destination.
Renvoie
Chemins des images présentes, triés.

◆ from_tensor()

aidge_in_space.images.from_tensor ( t,
mode )

Tenseur 1 x C x H x W dans [0, 1] -> tableau uint8 dans le mode demandé.

Paramètres
tTenseur (valeurs hors de [0, 1] écrêtées).
mode"gris" (une sortie RGB est convertie en luminance) ou "rgb".
Renvoie
Tableau uint8.

◆ to_tensor()

aidge_in_space.images.to_tensor ( img,
channels = None )

Tableau uint8 -> tenseur 1 x C x H x W dans [0, 1].

Paramètres
imgImage uint8, H x W ou H x W x 3.
channelsCanaux attendus par le modèle : une image grise est répliquée si 3.
Renvoie
Tenseur float32.