Revert "Optimized inference with onnx for patchcore. (#652)" (#791)

* Revert "Optimized inference with onnx for patchcore. (#652)" This reverts commit e965c8a. * update changelog
openvinotoolkit · Dec 14, 2022 · 32c2d19 · 32c2d19
1 parent 942cada
commit 32c2d19
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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 (<https://github.com/openvinotoolkit/anomalib/pull/791>)
 - Fix zero seed (<https://github.com/openvinotoolkit/anomalib/pull/766>)
 - Fix #699 (<https://github.com/openvinotoolkit/anomalib/pull/700>)
 - 🐞 Fix folder dataset for classification tasks (<https://github.com/openvinotoolkit/anomalib/pull/708>)

diff --git 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())