FCN (Fully Convolutional Network) in Computer Vision - Model Pipeline Trace

Model Pipeline - FCN (Fully Convolutional Network)

An FCN is a special type of neural network that looks at images and labels each pixel. It helps computers understand what parts of a picture belong to which object, like coloring each pixel as 'car' or 'road'.

Data Flow - 7 Stages

1Input Image

1 image x 256 height x 256 width x 3 channels→Raw image loaded for processing→1 image x 256 height x 256 width x 3 channels

A 256x256 color photo with red, green, blue channels

↓

2Convolutional Layers

1 x 256 x 256 x 3→Apply filters to detect edges and shapes→1 x 256 x 256 x 64

Feature maps highlighting edges and textures

↓

3Pooling Layers

1 x 256 x 256 x 64→Reduce image size to focus on important features→1 x 128 x 128 x 64

Smaller feature maps summarizing regions

↓

4More Convolutional Layers

1 x 128 x 128 x 64→Extract deeper features→1 x 128 x 128 x 128

Feature maps capturing complex shapes

↓

5Upsampling Layers

1 x 128 x 128 x 128→Increase spatial size to original image size→1 x 256 x 256 x 128

Upscaled feature maps aligned with input size

↓

6Final Convolution

1 x 256 x 256 x 128→Produce pixel-wise class scores→1 x 256 x 256 x 3

Scores for 3 classes per pixel (e.g., background, object1, object2)

↓

7Softmax Activation

1 x 256 x 256 x 3→Convert scores to probabilities per pixel→1 x 256 x 256 x 3

Probabilities summing to 1 for each pixel

Training Trace - Epoch by Epoch

Loss
1.2 |****
0.9 |***
0.7 |**
0.55|*
0.45| 
     Epochs -> 1 2 3 4 5

Epoch	Loss ↓	Accuracy ↑	Observation
1	1.2	0.45	Model starts learning basic features
2	0.9	0.60	Improved pixel classification
3	0.7	0.72	Model captures more details
4	0.55	0.80	Better segmentation boundaries
5	0.45	0.85	Model converging well

Prediction Trace - 7 Layers

Layer 1: Input Image

Layer 2: Convolutional Layer

Layer 3: Pooling Layer

Layer 4: More Convolutional Layers

Layer 5: Upsampling Layer

Layer 6: Final Convolution

Layer 7: Softmax Activation

Model Quiz - 3 Questions

What is the main purpose of the upsampling layers in an FCN?

A To apply activation functions like ReLU

B To reduce the number of channels in the feature maps

C To increase the spatial size of feature maps back to the input image size

D To split the image into smaller patches

Key Insight

Fully Convolutional Networks learn to label each pixel by combining convolution, pooling, and upsampling layers. Training improves the model by lowering loss and raising accuracy, helping it understand image details better.

Practice

1. What is the main advantage of a Fully Convolutional Network (FCN) compared to a traditional CNN for image tasks?

easy

A. It can label every pixel in the image, preserving spatial information.

B. It uses fully connected layers to improve classification accuracy.

C. It reduces the image size to a single vector for faster processing.

D. It only works with grayscale images for simplicity.

Solution

1. Step 1: Understand FCN structure

   FCNs replace fully connected layers with convolution layers to keep spatial info.

2. Step 2: Compare with traditional CNNs

   Traditional CNNs use fully connected layers that lose pixel location details, FCNs keep pixel-wise info.

3. Final Answer:

   It can label every pixel in the image, preserving spatial information. -> Option A

4. Quick Check:

   FCN labels pixels = B [OK]

Hint: FCNs keep pixel info; CNNs lose it with fully connected layers [OK]

Common Mistakes:

• Thinking FCNs use fully connected layers
• Assuming FCNs reduce image to vector
• Believing FCNs only work on grayscale

2. Which of the following is the correct way to replace a fully connected layer with a convolutional layer in an FCN?

easy

A. nn.Linear(in_features, out_features)

B. nn.ReLU()

C. nn.Conv2d(in_channels, out_channels, kernel_size=1)

D. nn.MaxPool2d(kernel_size=2)

Solution

1. Step 1: Identify layer types

   Fully connected layers are replaced by 1x1 convolution layers to keep spatial size.

2. Step 2: Match correct PyTorch layer

   nn.Conv2d with kernel size 1 acts like a fully connected layer per pixel.

3. Final Answer:

   nn.Conv2d(in_channels, out_channels, kernel_size=1) -> Option C

4. Quick Check:

   1x1 conv replaces FC layer = D [OK]

Hint: Use 1x1 conv to replace fully connected layers [OK]

Common Mistakes:

• Choosing nn.Linear which is fully connected
• Confusing pooling or activation layers as replacements
• Using wrong kernel sizes

3. Given this PyTorch FCN snippet, what is the output shape if the input is (1, 3, 64, 64)?

import torch
import torch.nn as nn

class SimpleFCN(nn.Module):
    def __init__(self):
        super().__init__()
        self.conv1 = nn.Conv2d(3, 16, 3, padding=1)
        self.conv2 = nn.Conv2d(16, 2, 1)
    def forward(self, x):
        x = self.conv1(x)
        x = self.conv2(x)
        return x

model = SimpleFCN()
input_tensor = torch.randn(1, 3, 64, 64)
output = model(input_tensor)
print(output.shape)

medium

A. (1, 2, 64, 64)

B. (1, 2, 62, 62)

C. (1, 16, 64, 64)

D. (1, 3, 64, 64)

Solution

1. Step 1: Analyze first conv layer

   Conv2d(3,16,3,padding=1) keeps spatial size 64x64, output shape (1,16,64,64).

2. Step 2: Analyze second conv layer

   Conv2d(16,2,1) with kernel size 1 keeps spatial size, output shape (1,2,64,64).

3. Final Answer:

   (1, 2, 64, 64) -> Option A

4. Quick Check:

   Padding keeps size; 1x1 conv keeps size = A [OK]

Hint: Padding keeps size; 1x1 conv keeps spatial dims [OK]

Common Mistakes:

• Ignoring padding effect on size
• Assuming kernel 3 reduces size without padding
• Confusing channel and spatial dimensions

4. You have an FCN model but the output size is smaller than the input image size. What is the most likely cause?

medium

A. Using 1x1 convolution layers increases spatial size.

B. Using convolution layers without padding reduces spatial size.

C. Using fully connected layers preserves spatial size.

D. Using batch normalization layers reduces spatial size.

Solution

1. Step 1: Understand convolution padding effect

   Convolution without padding shrinks spatial dimensions because kernel slides inside image.

2. Step 2: Identify cause of smaller output

   If output is smaller, likely padding is missing in conv layers causing shrinkage.

3. Final Answer:

   Using convolution layers without padding reduces spatial size. -> Option B

4. Quick Check:

   No padding shrinks output = A [OK]

Hint: No padding in conv layers shrinks output size [OK]

Common Mistakes:

• Thinking 1x1 conv changes size
• Believing batch norm affects size
• Assuming fully connected layers keep spatial size

5. You want to build an FCN for semantic segmentation on 128x128 images with 3 classes. Which architecture choice will produce output matching input size with class scores per pixel?

hard

A. Use fully connected layers after conv layers to classify whole image.

B. Use conv layers without padding and max pooling layers to reduce size.

C. Use conv layers with stride=2 and no upsampling layers.

D. Use conv layers with padding=1 and 1x1 conv at end, no pooling layers.

Solution

1. Step 1: Maintain spatial size

   Padding=1 with kernel size 3 keeps spatial size; no pooling avoids shrinking.

2. Step 2: Output class scores per pixel

   Use 1x1 conv with out_channels=3 to get class scores per pixel matching input size.

3. Final Answer:

   Use conv layers with padding=1 and 1x1 conv at end, no pooling layers. -> Option D

4. Quick Check:

   Padding + 1x1 conv = pixel-wise classes = C [OK]

Hint: Keep padding and no pooling for same size output [OK]

Common Mistakes:

• Using pooling reduces output size
• Using fully connected layers loses pixel info
• Using stride>1 without upsampling shrinks output