We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d78f995 commit e965c8aCopy full SHA for e965c8a
anomalib/models/patchcore/torch_model.py
@@ -38,7 +38,7 @@ def __init__(
38
self.num_neighbors = num_neighbors
39
40
self.feature_extractor = FeatureExtractor(backbone=self.backbone, pre_trained=pre_trained, layers=self.layers)
41
- self.feature_pooler = torch.nn.AvgPool2d(3, 1, 1)
+ self.feature_pooler = torch.nn.AvgPool2d(3, 1, 1, count_include_pad=False)
42
self.anomaly_map_generator = AnomalyMapGenerator(input_size=input_size)
43
44
self.register_buffer("memory_bank", Tensor())
0 commit comments