Skip to content
Snippets Groups Projects
Commit c1617a74 authored by Paul G's avatar Paul G
Browse files

entfernt

parent 0c546e47
No related branches found
No related tags found
No related merge requests found
%% Cell type:code id: tags:
```
###########################################################################
# ImageGenerator with dataframes
###########################################################################
#Size of our input images
SIZE = 128
# Size of ba
batch_size = 64
# Pfad zum Ordner, der nur Bilder der Klasse A enthält
src_path = "data/cell_images"
# Klasse "uninfected_train"
# Pfad zum Ordner mit den Bildern
src_path_train = "data/cell_images/uninfected_train"
# Liste der Dateinamen im Ordner
file_list_train = os.listdir(src_path_train)
# Liste der Labels (Klassen) für die Bilder
labels_train = ['uninfected_train'] * len(file_list_train) # Bilder im Ordner werden Klasse "uninfected_train" zugeordnet
# Erstellen eines DataFrames mit Dateinamen und den entsprechenden Labels
df_data_train = pd.DataFrame({'filename': file_list_train, 'label': labels_train})
# Klasse "uninfected_test"
src_path_test = "data/cell_images/uninfected_test"
file_list_test = os.listdir(src_path_test)
labels_test = ['uninfected_test'] * len(file_list_test)
df_data_test = pd.DataFrame({'filename': file_list_test, 'label': labels_test})
# Klasse "parasitized"
src_path_parasitized = "data/cell_images/parasitized"
file_list_parasitized = os.listdir(src_path_parasitized)
labels_parasitized = ['parasitized'] * len(file_list_parasitized)
df_data_parasitized = pd.DataFrame({'filename': file_list_parasitized, 'label': labels_parasitized})
#Define generators for training, validation and also anomaly data
# Konfigurieren des ImageDataGenerator für das Rescaling der Pixelwerte
datagen = ImageDataGenerator(rescale=1./255)
# Erstellen eines ImageDataGenerator-Objekts, um Bilder und Labels zu laden, Klasse "df_data_train"
train_generator = datagen.flow_from_dataframe(
df_data_train,
src_path_train, # Verzeichnis, das die Bilder enthält
x_col='filename', # Name der Spalte im DataFrame, die die Dateinamen enthält
y_col='label', # Name der Spalte im DataFrame, die die Labels enthält
target_size=(SIZE, SIZE), # Größe der Eingabebilder
batch_size=batch_size, # Anzahl der Bilder pro Batch
class_mode='categorical', # 'categorical' für Klassifikation, 'binary' für binäre Klassifikation
shuffle=True
)
validation_generator = datagen.flow_from_dataframe(
df_data_test,
src_path_test,
x_col='filename',
y_col='label',
target_size=(SIZE, SIZE),
batch_size=batch_size,
class_mode='categorical',
shuffle=True
)
anomaly_generator = datagen.flow_from_dataframe(
df_data_parasitized,
src_path_parasitized,
x_col='filename',
y_col='label',
target_size=(SIZE, SIZE),
batch_size=batch_size,
class_mode='categorical',
shuffle=True
)
```
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment