Skip to main content

🔍 The Convolution Operation

This is the magic that makes Computer Vision possible!

🔦 The Flashlight Analogy​

Imagine you are looking for a hidden message on a wall in a dark room. You take a small square flashlight (a Filter or Kernel) and slide it across the wall, checking a small 3x3 square at a time.

  • If the 3x3 square matches what your flashlight is looking for (e.g., a vertical edge), it lights up brightly!
  • If it doesn't match, it stays dark.

A Convolutional layer is just hundreds of these flashlights, scanning the image looking for edges, corners, and textures.

🐍 Python Implementation​

import torch
import torch.nn as nn

# 1 dummy image: 3 color channels, 64x64 pixels
# Shape: (Batch_Size, Channels, Height, Width)
dummy_image = torch.randn(1, 3, 64, 64)

# Create a Convolutional Layer!
# It expects 3 input channels, and will use 16 different "flashlights" (filters)
conv_layer = nn.Conv2d(in_channels=3, out_channels=16, kernel_size=3, padding=1)

# Scan the image!
output = conv_layer(dummy_image)

# The output now has 16 channels, one for each flashlight's findings!
print("Output shape:", output.shape) # (1, 16, 64, 64)