[PYTHON] Beschriften Sie Bilder im Jupiter-Labor

Was du machen willst

Ich möchte die Bilder beschriften, die ich für AI verwenden möchte, aber es ist mühsam, eine GUI zu erstellen. .. .. ich dachte Ich fragte mich, ob es mit JupyterLab möglich war, und nachdem ich verschiedene Dinge recherchiert hatte, konnte ich es tun, also ein Memo zu dieser Zeit. Ein Bild wie das Eingeben einer Beschriftung mit Eingabe, während das mit matplot angezeigte Bild aktualisiert wird.

Ausführungsumgebung

• Python 3.8.5
• jupyter lab 2.2.2 --mac (läuft bereits unter Windows 10)

Code

Beschriften Sie die Bilder im Ordner sample_img.

Vorbereitung

import cv2
import matplotlib.pyplot as plt
import os
import glob
import IPython
#Holen Sie sich den Pfad des Bildes
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 zur Anzeige des Bildes

name_list = []
for img_path in img_list:
    #Bilder laden
    img = cv2.imread(img_path)
    #Rahmen zur Anzeige des Bildes
    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)
    
    #Textfeld zum Beschriften
    comment = input()
   
    if comment == "break":
        break
    else:
        #Löschen Sie das angezeigte Bild
        IPython.display.clear_output()
        #Fügen Sie der Liste die in Eingabe eingegebene Zeichenfolge hinzu
        name_list.append(comment)

Der Bedienbildschirm sieht folgendermaßen aus. pass

Bestätigung

# name_Checkliste
name_list
>>> ['dog', 'cat', 'cat']

Schließlich

Ich wünschte, ich könnte den Eingangsteil wie ein Optionsfeld machen.