Python-Methode zum Lesen aller Bilder in einem angegebenen Ordner

Das Folgende ist eine Python-Methode zum Lesen aller Bilder in einem bestimmten Ordner. Sie hat einen guten Referenzwert und ich hoffe, dass sie für alle hilfreich ist. Schauen wir uns das gemeinsam an

(1) Datenvorbereitung

Einführung in den Datensatz:

Der Datensatz speichert 1223 dort Es gibt 756 negative Proben (Bildnamen sind 0,1–0,756) und 458 positive Proben (Bildnamen sind 1,1–1,458), darunter: Die Beschriftung vor „.“ ist die Probenbezeichnung und die Beschriftung nach „.“ ist die Probenserie Nummer

(2) Verwenden Sie Python, um alle Bilder im Ordner zu lesen

'''
Load the image files form the folder
input:
  imgDir: the direction of the folder
  imgName:the name of the folder
output:
  data:the data of the dataset
  label:the label of the datset
'''
def load_Img(imgDir,imgFoldName):
  imgs = os.listdir(imgDir+imgFoldName)
  imgNum = len(imgs)
  data = np.empty((imgNum,1,12,12),dtype="float32")
  label = np.empty((imgNum,),dtype="uint8")
  for i in range (imgNum):
    img = Image.open(imgDir+imgFoldName+"/"+imgs[i])
    arr = np.asarray(img,dtype="float32")
    data[i,:,:,:] = arr
    label[i] = int(imgs[i].split('.')[0])
  return data,label

Die hier erhaltenen Daten und Beschriftungen sind beide Ndarray-Daten

Daten: (1223,1,12,12)

Beschriftung: (1223,)

Hinweis: Der Datentyp nddary ist ein von Numpy bereitgestellter Datentyp, also ein N-dimensionales Array, das den Fehler ausgleicht, dass Arrays in Python keine Mehrdimensionalität unterstützen

(3) Aufrufmethode

craterDir = "./data/CraterImg/Adjust/"
foldName = "East_CraterAdjust12"
data, label = load_Img(craterDir,foldName)

Verwandte Empfehlungen:

Ein Beispiel dafür, wie Python eine CSV-Datei liest und die Datei in eine Liste einfügt

Python implementiert die Methode zum Generieren einer TXT-Datei mit Beschriftungen für Bilder im Datei

Das obige ist der detaillierte Inhalt vonPython-Methode zum Lesen aller Bilder in einem angegebenen Ordner. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!