Downloads · 30 days
0
Nebrass1/image_captioning_resnet_lstm
image_captioning_resnet_lstm is a image-to-text model from Nebrass1. Use it when you need a caption or text from an image. It is set up for keras. The card lists the license as mit.
Ce modèle a été entraîné sur un sous-ensemble du jeu de données Flickr30k. Il utilise une architecture Codeur-Décodeur (ResNet50 pour l'extraction des caractéristiques de l'image et un LSTM pour la génération séquenti…
Downloads · 30 days
0
Access
Public
Updated Oct 22, 2025
Repo size
60.1 MB
Likes
0
Public
Click a slice to open those files.
.h559.8 MB · 99%
From the Hugging Face model README
Ce modèle a été entraîné sur un sous-ensemble du jeu de données Flickr30k. Il utilise une architecture Codeur-Décodeur (ResNet50 pour l'extraction des caractéristiques de l'image et un LSTM pour la génération séquentielle de texte).
Objectifs : Le modèle prend une image en entrée et génère une description textuelle.
Artefacts essentiels:
caption_decoder_final.weights.h5: Poids entraînés du décodeur.tokenizer.pkl: Tokenizer pour la conversion mot/ID.model.py: Fichier d'architecture du modèle.ATTENTION: Le modèle peut présenter un biais d'apprentissage ("man on the beach") dû à l'utilisation du vocabulaire non filtré. Une nouvelle session d'entraînement est prévue.