Transfer Learning Model

21/03/2024, 21:20 Untitled17.
ipynb - Colaboratory
from google.colab import drive

drive.mount('/content/drive')
Mounted at /content/drive
Double-click (or enter) to edit
import tensorflow as tf
from tensorflow.keras.preprocessing.image import ImageDataGenerator
from tensorflow.keras.applications import MobileNetV2
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense, GlobalAveragePooling2D
Start coding or generate with AI.
Start coding or generate with AI.
keyboard_arrow_down Define data directories

train_data_dir = "/content/drive/MyDrive/data/Snake Images/train"
test_data_dir = "/content/drive/MyDrive/data/Snake Images/test"
# Image size and batch size

img_height, img_width = 224, 224
batch_size = 32
# Use ImageDataGenerator to load the data and perform data augmentation

train_datagen = ImageDataGenerator(
rescale=1./255,
shear_range=0.2,
zoom_range=0.2,
horizontal_flip=True,
validation_split=0.2) # 80% for training, 20% for validation
test_datagen = ImageDataGenerator(rescale=1./255)
train_generator = train_datagen.flow_from_directory(
train_data_dir,
target_size=(img_height, img_width),
batch_size=batch_size,
class_mode='categorical',
subset='training') # Use subset 'training' for training data
Found 1420 images belonging to 2 classes.
validation_generator = train_datagen.flow_from_directory(
train_data_dir,
class_mode='categorical',
subset='validation') # Use subset 'validation' for validation data
test_generator = test_datagen.flow_from_directory(
test_data_dir,
class_mode='categorical')
https://colab.research.google.com/drive/107zbX3NrHW8LuMC7Zgs8IwvE7euH5MLE#scrollTo=tI0Tb9kHSdDb&printMode=true 1/4
21/03/2024, 21:20 Untitled17.ipynb - Colaboratory
# Load MobileNetV2 pre-trained model
base_model = MobileNetV2(weights='imagenet', include_top=False, input_shape=(img_height, img_width, 3))
# Freeze the base model layers

for layer in base_model.layers:
layer.trainable = False
Downloading data from https://storage.googleapis.com/tensorflow/keras-applications/mobilenet_v2/mobilenet_v2_weights_tf_dim_ordering

9406464/9406464 [==============================] - 0s 0us/step
# Create the model

model = Sequential([
base_model,
GlobalAveragePooling2D(),
Dense(256, activation='relu'),
Dense(train_generator.num_classes, activation='softmax')
])
# Compile the model

model.compile(optimizer='adam',
loss='categorical_crossentropy',
metrics=['accuracy'])
# Train the model

model.fit(
train_generator,
steps_per_epoch=train_generator.samples // batch_size,
validation_data=validation_generator,
validation_steps=validation_generator.samples // batch_size,
epochs=10)
Epoch 1/10
44/44 [==============================] - 599s 14s/step - loss: 0.6070 - accuracy: 0.7147 - val_loss: 0.4611 - val_accuracy: 0.7699
Epoch 2/10
44/44 [==============================] - 31s 710ms/step - loss: 0.3612 - accuracy: 0.8393 - val_loss: 0.3795 - val_accuracy: 0.8125
Epoch 3/10
Epoch 4/10
Epoch 5/10
Epoch 6/10
Epoch 7/10
Epoch 8/10
Epoch 9/10
Epoch 10/10
<keras.src.callbacks.History at 0x7d795ffaa830>
# Evaluate the model on test data

test_loss, test_acc = model.evaluate(test_generator, verbose=2)
print('\nTest accuracy:', test_acc)
9/9 - 93s - loss: 0.3970 - accuracy: 0.8290 - 93s/epoch - 10s/step
Test accuracy: 0.8289963006973267
# Train the model and store the history

history = model.fit(
train_generator,
steps_per_epoch=train_generator.samples // batch_size,
validation_data=validation_generator,
validation_steps=validation_generator.samples // batch_size,
epochs=10)
# Get training accuracy from history

training_accuracy = history.history['accuracy'][-1]
print('\nTraining accuracy:', training_accuracy)
Epoch 1/10
Epoch 2/10
Epoch 3/10
Epoch 4/10
Epoch 5/10
Epoch 6/10
Epoch 7/10
Epoch 8/10
Epoch 9/10
Epoch 10/10
Training accuracy: 0.990634024143219
import matplotlib.pyplot as plt
# Get the first batch of images and labels from the training set
images, labels = next(train_generator)
# Display the first nine images and labels

plt.figure(figsize=(20, 20))
for i in range(19):
ax = plt.subplot(6, 6, i + 1)
plt.imshow(images[i])
plt.title(f'Label: {labels[i]}')
plt.axis('off')
plt.show()

Transfer Learning Model

Uploaded by

Document Information

Original Description:

Original Title

Copyright

Available Formats

Share this document

Share or Embed Document

Sharing Options

Did you find this document useful?

Is this content inappropriate?

Copyright:

Available Formats

Transfer Learning Model

Uploaded by

Copyright:

Available Formats

21/03/2024, 21:20 Untitled17.

from google.colab import drive

Double-click (or enter) to edit

Start coding or generate with AI.

Start coding or generate with AI.

keyboard_arrow_down Define data directories

# Image size and batch size

# Use ImageDataGenerator to load the data and perform data augmentation

Found 1420 images belonging to 2 classes.

Found 355 images belonging to 2 classes.

Found 269 images belonging to 2 classes.

# Freeze the base model layers

Downloading data from https://storage.googleapis.com/tensorflow/keras-applications/mobilenet_v2/mobilenet_v2_weights_tf_dim_ordering

# Create the model

# Compile the model

# Train the model

# Evaluate the model on test data

9/9 - 93s - loss: 0.3970 - accuracy: 0.8290 - 93s/epoch - 10s/step

Test accuracy: 0.8289963006973267

# Train the model and store the history

# Get training accuracy from history

Training accuracy: 0.990634024143219

import matplotlib.pyplot as plt

# Display the first nine images and labels

You might also like