momotoculteur / multi-label-image-classification

Permet de creer avec Tensorflow et Keras un recommandeur de hastag pour image, avec des algorithmes de deep learning.

Recommendeur de hashtag pour image

Permet de creer avec Tensorflow et Keras un recommandeur de hastag pour image, avec des algorithmes de deep learning.

Installer les pré-requis

Je vous laisse installer vous même les libraires nécessaires, par conda ou pip selon vos préferences et environnements

• Tensorflow
• Keras
• Pandas
• TQDM
• PIL

Téléchargez le dataset HARRISON, le dézipper dans ./HARRISSON/.

Customization des hyper paramètres

Disponible dans le fichier trainModel.py :

Attribut	Description
CLASSE	Lien vers le fichier contenant la liste totales des classes avec leurs indices respectifs
NB_CLASSES	Nombre de classe ; définie le nombre de neuronnes sur le dernier layer
NB_EPOCH	Nombre d'iteration pour l'entrainement
BATCH_SIZE	Nombre d'image par lot
SHUFFLE	Permet de melanger notre dataset
IMG_SIZE	Taille des images à resize
TRAINSIZE_RATIO	Ratio pour gerer la taille des jeux d'entrainement et de validation
DIRECTORY_DATA	Chemin parent vers les images
DIRECTORY_TRAINED_MODEL	Chemin ou save le modele
COLOR_MODE	Choix des canaux des images : couleur ou noir/blanc

Générer et entrainer un model

$ python trainModel.py

Prédiction de nouvelles images

Donnez les chemins d'accès à vos photos tests dans le fichier ./dataTest/imgTest.txt, puis lancez :

$ python autoPredic.py