Skip to content

Does this show overfitting? #537

@Billy-ZTB

Description

@Billy-ZTB

I used a VGG16 based UNet to learn to segment buildings from an image. During training, I found that the encoder's activation starts to shrink to small area of the buildings, and also show up on some background area, does this show that my model is overfitting?

Examples from epoch 5 and epoch 24:

Epoch5:
image2_heatmap
Epoch 24:
image2_heatmap

Epoch5:
image5_heatmap
Epoch24:
image5_heatmap

Epoch5:
image12_heatmap
Epoch24:
image12_heatmap

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions