[PYTHON] Étiqueter des images sur Jupyter Lab

Chose que tu veux faire

Je souhaite étiqueter les images que je souhaite utiliser pour l'IA, mais il est difficile de créer une interface graphique. .. .. j'ai pensé Je me suis demandé si cela pouvait être fait sur JupyterLab, et j'ai fait beaucoup de recherches, alors j'ai pris une note à ce moment-là. Une image comme la saisie d'étiquettes avec entrée lors de la mise à jour de l'image affichée avec matplot.

Environnement d'exécution

• Python 3.8.5
• jupyter lab 2.2.2 --mac (déjà exécuté sous Windows 10)

code

Étiquetez les images dans le dossier sample_img.

Préparation

import cv2
import matplotlib.pyplot as plt
import os
import glob
import IPython
#Obtenez le chemin de l'image
img_list = glob.glob(os.path.join(r"sample_img","*.jpg "))
img_list
>>> ['sample_img/img3.jpg', 'sample_img/img2.jpg', 'sample_img/img1.jpg']

Code pour afficher l'image

name_list = []
for img_path in img_list:
    #Chargement des images
    img = cv2.imread(img_path)
    #Cadre pour afficher l'image
    fig = plt.figure(figsize=(5,5))
    ax = fig.add_subplot(1,1,1)
    ax.imshow(cv2.cvtColor(img,cv2.COLOR_BGR2RGB))
    plt.title(img_path)
    plt.pause(.01)
    
    #Zone de texte pour l'étiquetage
    comment = input()
   
    if comment == "break":
        break
    else:
        #Effacer l'image affichée
        IPython.display.clear_output()
        #Ajouter la chaîne de caractères saisie dans Input à la liste
        name_list.append(comment)

L'écran de fonctionnement ressemble à ceci. pass

Vérification

# name_Liste de contrôle
name_list
>>> ['dog', 'cat', 'cat']

finalement

J'aimerais pouvoir faire de la partie Entrée comme un bouton radio.