Scribd
Upload
4 views3 pages

Xception Model for Image Classification

The document outlines a TensorFlow script for training a deep learning model using the Xception architecture for image classification with five classes. It includes data preparation, model building, training, evaluation, and performance metrics calculation such as accuracy, precision, recall, and F1 score. The script leverages data augmentation and pre-trained weights to enhance model performance.

Uploaded by

Bhupesh Raj
Copyright
© All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as DOCX, PDF, TXT or read online on Scribd
4 views3 pages

Xception Model for Image Classification

The document outlines a TensorFlow script for training a deep learning model using the Xception architecture for image classification with five classes. It includes data preparation, model building, training, evaluation, and performance metrics calculation such as accuracy, precision, recall, and F1 score. The script leverages data augmentation and pre-trained weights to enhance model performance.

Uploaded by

Bhupesh Raj
Copyright
© All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as DOCX, PDF, TXT or read online on Scribd

import os

import tensorflow as tf
from [Link] import ImageDataGenerator
from [Link] import Xception
from [Link] import Sequential
from [Link] import Dense, Flatten, Dropout, GlobalAveragePooling2D
from [Link] import Adam
from [Link] import ModelCheckpoint
from [Link] import classification_report, confusion_matrix
import numpy as np

# Define constants
IMAGE_SIZE = 224
BATCH_SIZE = 64
NUM_CLASSES = 5
EPOCHS = 50

# Define data directories


base_dir = [Link]([Link](), 'The Gems Atlas')
train_dir = [Link](base_dir, 'train')
valid_dir = [Link](base_dir, 'valid')
test_dir = [Link](base_dir, 'test')

# Define data generators


train_datagen = ImageDataGenerator( rescale=1./255,
shear_range=0.2, zoom_range=0.2,
horizontal_flip=True, validation_split=0.2)
train_generator = train_datagen.flow_from_directory( train_dir,
target_size=(IMAGE_SIZE, IMAGE_SIZE), batch_size=BATCH_SIZE,
class_mode='categorical',
subset='training')

validation_generator = train_datagen.flow_from_directory( train_dir,


target_size=(IMAGE_SIZE, IMAGE_SIZE), batch_size=BATCH_SIZE,
class_mode='categorical', subset='validation')
test_datagen = ImageDataGenerator(rescale=1./255)

test_generator = test_datagen.flow_from_directory(
test_dir,
target_size=(IMAGE_SIZE, IMAGE_SIZE), batch_size=BATCH_SIZE,
class_mode='categorical',
shuffle=False)

# Load pre-trained Inception V3 model


base_model = Xception(include_top=False, weights='imagenet', input_shape=(224, 224, 3))
# Freeze the layers in the base model
for layer in base_model.layers: [Link] = False
# Build the model
model = Sequential()
[Link](base_model)
[Link](Flatten())
[Link](Dense(256, activation='relu'))
[Link](Dropout(0.5))
[Link](Dense(NUM_CLASSES, activation='softmax'))

# Compile the model


[Link](optimizer=Adam(lr=0.001),
loss='categorical_crossentropy', metrics=['accuracy'])
# Train the model
history = [Link](
train_generator,
epochs=EPOCHS,
validation_data=validation_generator)

# Evaluate the model on the test data


loss, accuracy = [Link](test_generator)
print('Test loss:', loss)
print('Test accuracy:', accuracy)

y_pred = [Link](test_generator)
y_pred_classes = [Link](y_pred, axis=1)
y_true = test_generator.classes
# Print the confusion matrix
print(confusion_matrix(y_true, y_pred_classes)) # Print the classification report
target_names = list(test_generator.class_indices.keys())
print(classification_report(y_true, y_pred_classes, target_names=target_names)) # Calculate the
accuracy
accuracy = [Link](y_pred_classes == y_true)
print('Accuracy:', accuracy)
# Calculate the precision, recall, and F1 score
report = classification_report(y_true, y_pred_classes, target_names=target_names, output_dict=True)
precision = report['weighted avg']['precision']
recall = report['weighted avg']['recall']
f1_score = report['weighted avg']['f1-score']
print('Precision:', precision)
print('Recall:', recall)
print('F1 score:', f1_score)

You might also like