Envelopper et restituer du texte multiligne sur des images à l'aide de la bibliothèque Pillow de Python

Traitement d'image Python : la bibliothèque Pillow implémente l'annotation automatique du texte de retour à la ligne

Python est devenu le langage de programmation leader dans le domaine du traitement d'images avec ses riches bibliothèques open source. Pillow est l'une des bibliothèques de traitement d'images couramment utilisées. Elle est simple, facile à utiliser et dispose d'une documentation complète. Elle est souvent utilisée pour des opérations telles que la mise à l'échelle de l'image, le recadrage, le réglage de la luminosité et l'annotation.

Cependant, Pillow a un problème avec l'annotation du texte : lorsque le texte dépasse la largeur de la zone de texte, il ne s'enroulera pas automatiquement. La bibliothèque Pillow elle-même ne fournit pas cette fonction et nous devons écrire nous-mêmes l'implémentation logique.

Ce didacticiel montrera comment utiliser la bibliothèque Pillow pour ajouter une zone de texte de retour à la ligne en Python afin d'obtenir une annotation correcte du texte de l'image. L'effet final est le suivant :

L'image ci-dessus est une capture d'écran de mon profil Dev.to, nous l'utiliserons comme exemple pour expliquer. La zone de texte verte est l'annotation de texte que nous avons ajoutée.

Préparation

Ce tutoriel nécessite que vous ayez des connaissances de base en programmation Python, telles que les instructions conditionnelles (if, else), les boucles for, etc. Vous aurez également besoin des outils et logiciels suivants :

1. Python3  : Interpréteur pour exécuter des scripts Python.
2. Pillow : bibliothèque de traitement d'images Python.
3. Éditeur de code : tel que Pycharm, VScode, etc.

Créer un nouveau projet

Suivez ces étapes pour créer un nouveau projet :

A. Créez un nouveau dossier à l'aide du terminal/ligne de commande :

mkdir image_annotation

B. Utilisez pip pour installer virtualenv (ignorez cette étape si vous l'avez déjà installé) :

pip install virtualenv

C. Basculez le répertoire de travail vers le dossier image_annotation :

cd image_annotation

D. Créer un nouvel environnement virtuel :

virtualenv env

E. Activer l'environnement virtuel (utiliser l'invite de commande pour Windows) :

Windows :

.\env\Scripts\activate

Linux/macOS :

source env/bin/activate

F. Utilisez pip pour installer la bibliothèque Pillow :

pip install pillow

Ouvrez le projet dans l'éditeur de code et créez un nouveau fichier Python nommé script.py dans le dossier du projet.

Préparer l'image de base

L'image que vous souhaitez annoter est l'image de base. Ouvrez et préparez l'image à l'aide du module ImageDraw de Pillow. Écrivez le code suivant dans le fichier script.py :

from PIL import Image, ImageDraw, ImageFont
image_file = "path_to_image"  # 请替换为您的图片路径

# 打开图像
image = Image.open(image_file)

# 初始化ImageDraw
draw = ImageDraw.Draw(image)

Ajouter une annotation d'image

Pillow peut ajouter du texte brut et des zones de texte avec un remplissage en arrière-plan. Le texte peut être une seule ligne ou plusieurs lignes. Ce didacticiel explique comment ajouter une zone de texte multiligne.

La méthode

ImageDraw.multiline_text() peut ajouter plusieurs lignes de texte brut, mais pas de remplissage d'arrière-plan. La méthode ImageDraw.rectangle() peut ajouter une zone de texte avec un remplissage en arrière-plan.

Ajoutez le code suivant dans le fichier script.py :

mkdir image_annotation

Ce code définit le texte, la police et la largeur de la zone de texte. Les variables x et y représentent le point de départ du dessin, et end_x et end_y représentent les coordonnées du coin inférieur droit de la zone de texte. La largeur et la hauteur de la zone de texte sont respectivement de 200 et 50.

Les méthodes

ImageDraw.rectangle() et ImageDraw.multiline_text() sont utilisées respectivement pour dessiner des zones de texte et du texte multiligne. La méthode image.show() est utilisée pour afficher l'image traitée. Vous pouvez enregistrer l'image en utilisant image.save("new_image.png"). Les résultats sont les suivants :

Il y a toujours un problème avec l'annotation dans l'image ci-dessus, et le texte multiligne ne s'enroule pas automatiquement. La section suivante explique comment résoudre ce problème.

Réaliser un retour à la ligne automatique

Le caractère de saut de ligne n est utilisé pour spécifier la position du saut de ligne. Dans l'exemple précédent, le contenu après le caractère de nouvelle ligne n sera renvoyé à la ligne. Mais dans les applications pratiques, la longueur du texte est généralement dynamique et il est difficile de déterminer la position du caractère de nouvelle ligne.

L'attribut

du module ImageDraw de .textlength()Pillow peut calculer la longueur du texte et la comparer avec la largeur de la zone de texte pour déterminer la position du saut de ligne.

Créez une nouvelle fonction nommée script.py en haut du fichier wrap_text() (après l'instruction d'importation), contenant la logique de retour à la ligne :

pip install virtualenv

Ajoutez le code suivant après les variables text, font, max_width :

cd image_annotation

Remplacez la méthode draw.multiline_text() par le code suivant :

virtualenv env

Supprimez les nouvelles lignes du texte n et exécutez le code :

.\env\Scripts\activate

Le résultat affiché montre que le texte dépasse toujours la hauteur de la zone de texte. Tandis que le texte s'ajuste automatiquement à la largeur de la zone de texte, la hauteur de la zone de texte est fixe, provoquant un débordement du texte.

Définir la hauteur de la zone de texte dynamique

La hauteur de la zone de texte dynamique est déterminée en fonction du nombre de lignes de texte. La première étape consiste à changer la variable end_y de la zone de texte en une valeur dynamique :

source env/bin/activate

Cette formule a été trouvée après de nombreuses expériences et elle semble être la meilleure solution pour obtenir une hauteur de zone de texte dynamique dans ce cas d'utilisation. wrapped_linesLa liste contient toutes les lignes à ajouter à la zone de texte, la longueur de la liste est donc égale au nombre total de lignes de la zone de texte.

Les résultats sont les suivants :

Vous devrez peut-être multiplier le nombre total de lignes par différentes valeurs pour obtenir la solution parfaite pour votre cas d'utilisation.

Ajouter un remplissage de texte

Le texte est trop proche du bord de la zone de texte, ce qui affecte la lisibilité et le style. Vous pouvez résoudre ce problème en ajoutant un remplissage à l'intérieur de la zone de texte. Ajoutez une nouvelle variable script.py dans le fichier padding et modifiez la taille de la zone de texte :

pip install pillow

Ce code permet un espacement entre le texte et les bords de la zone de texte.

Ajouter un pointeur

Le pointeur peut facilement indiquer la partie de l'image à laquelle l'annotation/l'étiquette fait référence. Le pointeur doit être avant l'étiquette. Cela signifie que le pointeur sera dessiné à la position actuelle de la zone de texte et que la zone de texte se déplacera vers la droite.

Par conséquent, l'axe des x de la zone de texte sera associé à la nouvelle variable box_x. Ce changement doit également être reflété dans d'autres variables en utilisant l'axe X de la zone de texte. Voici le code mis à jour :

mkdir image_annotation

Dans le code ci-dessus, la méthode ImageDraw.circle() (où 10 est le rayon) est utilisée pour dessiner le pointeur au point spécifié. box_xLa variable est la nouvelle valeur de l'axe des x de la zone de texte.

Code complet

Ce qui suit est le code complet du script.py fichier :

pip install virtualenv

Conclusion

Le traitement des images n'est pas toujours aussi difficile qu'il y paraît. Bien que certaines bibliothèques de traitement d'images ne puissent pas résoudre directement votre problème avec leurs modules, vous pouvez utiliser les modules existants pour implémenter une solution spécifique à votre cas d'utilisation. C'est la beauté du codage : être capable de résoudre des problèmes avec des solutions personnalisées et spécifiques.

Dans ce didacticiel, vous avez appris à utiliser la bibliothèque Pillow de Python pour annoter des images, ajouter des zones de texte multilignes renvoyées à la ligne, et bien plus encore. Vous avez également appris à écrire des formules mathématiques qui peuvent vous aider dans le traitement des images.

Veuillez vous référer à la documentation Pillow pour plus de détails sur les modules utilisés.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!