DNN-based face detection in Computer Vision - ML Experiment: Train & Evaluate

Experiment - DNN-based face detection

Problem:Detect faces in images using a deep neural network (DNN). The current model is trained on a small dataset and achieves high training accuracy but low validation accuracy.

Current Metrics:Training accuracy: 98%, Validation accuracy: 70%, Training loss: 0.05, Validation loss: 0.45

Issue:The model is overfitting: it performs very well on training data but poorly on validation data.

Your Task

Reduce overfitting so that validation accuracy improves to at least 85% while keeping training accuracy below 92%.

You can only modify the model architecture and training hyperparameters.

Do not change the dataset or add new data.

Keep the input image size and preprocessing steps the same.

Hint 1

Hint 2

Hint 3

Hint 4

Solution

Computer Vision

import tensorflow as tf
from tensorflow.keras import layers, models

# Define the improved model with dropout and batch normalization
model = models.Sequential([
    layers.Input(shape=(64, 64, 3)),
    layers.Conv2D(32, (3, 3), activation='relu'),
    layers.BatchNormalization(),
    layers.MaxPooling2D((2, 2)),
    layers.Dropout(0.25),

    layers.Conv2D(64, (3, 3), activation='relu'),
    layers.BatchNormalization(),
    layers.MaxPooling2D((2, 2)),
    layers.Dropout(0.25),

    layers.Flatten(),
    layers.Dense(64, activation='relu'),
    layers.BatchNormalization(),
    layers.Dropout(0.5),
    layers.Dense(1, activation='sigmoid')
])

model.compile(optimizer=tf.keras.optimizers.Adam(learning_rate=0.0005),
              loss='binary_crossentropy',
              metrics=['accuracy'])

# Assume X_train, y_train, X_val, y_val are preloaded datasets
# Train the model with early stopping
early_stop = tf.keras.callbacks.EarlyStopping(monitor='val_loss', patience=5, restore_best_weights=True)

history = model.fit(
    X_train, y_train,
    epochs=50,
    batch_size=32,
    validation_data=(X_val, y_val),
    callbacks=[early_stop]
)

Added dropout layers after convolutional and dense layers to reduce overfitting.

Added batch normalization layers to stabilize and speed up training.

Reduced the number of neurons in the dense layer from 128 to 64 to simplify the model.

Lowered the learning rate to 0.0005 for smoother convergence.

Added early stopping to prevent over-training.

Results Interpretation

Before: Training accuracy 98%, Validation accuracy 70%, Training loss 0.05, Validation loss 0.45

After: Training accuracy 90%, Validation accuracy 87%, Training loss 0.20, Validation loss 0.30

Adding dropout and batch normalization, simplifying the model, and using early stopping helped reduce overfitting. This improved validation accuracy while keeping training accuracy reasonable, showing better generalization to new data.

Bonus Experiment

Try using data augmentation techniques to artificially increase the training data variety and see if validation accuracy improves further.

💡 Hint

Use image transformations like rotation, flipping, and zooming during training to help the model learn more robust features.

Practice

1. What is the main purpose of using a DNN-based face detection model?

easy

A. To increase the size of an image

B. To find faces in images automatically

C. To compress images for storage

D. To change the color of an image

Solution

1. Step 1: Understand the goal of face detection

   Face detection means finding where faces are in pictures or videos.

2. Step 2: Recognize what DNN-based models do

   DNN models learn patterns to automatically locate faces without manual rules.

3. Final Answer:

   To find faces in images automatically -> Option B

4. Quick Check:

   Face detection = find faces [OK]

Hint: Face detection means locating faces in images [OK]

Common Mistakes:

• Confusing detection with image editing
• Thinking it changes image colors
• Mixing detection with compression

2. Which OpenCV function is used to load a pre-trained DNN face detection model?

easy

A. cv2.dnn.readNet()

B. cv2.imread()

C. cv2.resize()

D. cv2.cvtColor()

Solution

1. Step 1: Identify the function for loading DNN models

   OpenCV's DNN module uses cv2.dnn.readNet() to load models from files.

2. Step 2: Differentiate from other functions

   cv2.imread() loads images, cv2.resize() changes size, cv2.cvtColor() changes color space.

3. Final Answer:

   cv2.dnn.readNet() -> Option A

4. Quick Check:

   Load DNN model = cv2.dnn.readNet() [OK]

Hint: Use cv2.dnn.readNet() to load DNN models [OK]

Common Mistakes:

• Using cv2.imread() to load models
• Confusing image functions with model loading
• Trying to resize models with cv2.resize()

3. Given this code snippet, what will be the output type of detections?

net = cv2.dnn.readNet('model.caffemodel', 'deploy.prototxt')
blob = cv2.dnn.blobFromImage(image, 1.0, (300, 300), (104, 177, 123))
net.setInput(blob)
detections = net.forward()

medium

A. A list of bounding box coordinates

B. A single integer count of faces

C. A string describing detected faces

D. A NumPy array with detection data

Solution

1. Step 1: Understand net.forward() output

   net.forward() returns a NumPy array containing detection results with scores and box info.

2. Step 2: Differentiate output types

   It is not a list, integer, or string but a multi-dimensional array with detection details.

3. Final Answer:

   A NumPy array with detection data -> Option D

4. Quick Check:

   net.forward() output = NumPy array [OK]

Hint: net.forward() returns a NumPy array of detections [OK]

Common Mistakes:

• Thinking output is a list or count
• Expecting a string description
• Confusing output with input image

4. You run this code but get no faces detected. What is the likely error?

blob = cv2.dnn.blobFromImage(image, 1.0, (224, 224), (104, 177, 123))
net.setInput(blob)
detections = net.forward()

What should you fix?

medium

A. Change input size to (300, 300) to match model requirements

B. Remove mean subtraction values from blobFromImage

C. Use cv2.imread() instead of blobFromImage

D. Call net.forward() before setInput()

Solution

1. Step 1: Check model input size requirements

   Many DNN face detectors expect input size (300, 300). Using (224, 224) causes mismatch and no detection.

2. Step 2: Correct the input size in blobFromImage

   Changing to (300, 300) matches model and enables detection.

3. Final Answer:

   Change input size to (300, 300) to match model requirements -> Option A

4. Quick Check:

   Input size must match model = (300, 300) [OK]

Hint: Match blob size to model input size [OK]

Common Mistakes:

• Using wrong input size for the model
• Calling net.forward() before setInput()
• Removing mean subtraction incorrectly

5. You want to improve face detection accuracy on images with multiple faces at different sizes. Which approach helps most?

hard

A. Reduce the input image size to speed up detection

B. Increase the learning rate during training

C. Use image pyramid to detect faces at multiple scales

D. Remove mean subtraction from preprocessing

Solution

1. Step 1: Understand multi-scale detection challenge

   Faces appear at different sizes; detecting all requires checking multiple scales.

2. Step 2: Use image pyramid technique

   Image pyramid resizes image multiple times to detect faces at various sizes, improving accuracy.

3. Final Answer:

   Use image pyramid to detect faces at multiple scales -> Option C

4. Quick Check:

   Multi-scale detection = image pyramid [OK]

Hint: Use image pyramid for faces at different sizes [OK]

Common Mistakes:

• Thinking higher learning rate fixes scale issues
• Reducing image size loses small faces
• Skipping mean subtraction harms accuracy