diff --git a/CHANGELOG.md b/CHANGELOG.md index 6fa0989e59..16198b29c7 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -30,6 +30,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/). ### Fixed +- Fix PatchCore performance deterioration by reverting changes to Average Pooling layer () - Fix zero seed () - Fix #699 () - 🐞 Fix folder dataset for classification tasks () diff --git a/anomalib/models/patchcore/torch_model.py b/anomalib/models/patchcore/torch_model.py index db9a814166..819fd83945 100644 --- a/anomalib/models/patchcore/torch_model.py +++ b/anomalib/models/patchcore/torch_model.py @@ -38,7 +38,7 @@ def __init__( self.num_neighbors = num_neighbors self.feature_extractor = FeatureExtractor(backbone=self.backbone, pre_trained=pre_trained, layers=self.layers) - self.feature_pooler = torch.nn.AvgPool2d(3, 1, 1, count_include_pad=False) + self.feature_pooler = torch.nn.AvgPool2d(3, 1, 1) self.anomaly_map_generator = AnomalyMapGenerator(input_size=input_size) self.register_buffer("memory_bank", Tensor())