!2190 amend_patchcore

Merge pull request !2190 from jiangzhenguang/amend_patchcore

official/cv/patchcore/preprocess.py

@@ -108,7 +108,7 @@ def createDataset(dataset_path, category):
                              transform=data_transforms, gt_transform=gt_transforms, phase='test')
 
     train_dataset = ds.GeneratorDataset(train_data, column_names=['img', 'gt', 'label', 'idx'],
-                                        shuffle=True)
+                                        shuffle=False)
     test_dataset = ds.GeneratorDataset(test_data, column_names=['img', 'gt', 'label', 'idx'],
                                        shuffle=False)
 

official/cv/patchcore/src/dataset.py

@@ -156,7 +156,7 @@ def createDataset(dataset_path, category):
                              transform=data_transforms, gt_transform=gt_transforms, phase='test')
 
     train_dataset = ds.GeneratorDataset(train_data, column_names=['img', 'gt', 'label', 'idx'],
-                                        shuffle=True)
+                                        shuffle=False)
     test_dataset = ds.GeneratorDataset(test_data, column_names=['img', 'gt', 'label', 'idx'],
                                        shuffle=False)
 

official/cv/patchcore/src/model.py

@@ -78,7 +78,7 @@ class ResNet(nn.Cell):
         self.conv1 = nn.Conv2d(3, self.inplanes, kernel_size=7, stride=2, padding=3, pad_mode='pad')
         self.bn1 = nn.BatchNorm2d(self.inplanes)
         self.relu = nn.ReLU()
-        self.pad = nn.Pad(paddings=((0, 0), (0, 0), (1, 1), (1, 1)))
+        self.pad = nn.Pad(paddings=((0, 0), (0, 0), (1, 1), (1, 1)), mode="REFLECT")
         self.maxpool = nn.MaxPool2d(kernel_size=3, stride=2, pad_mode="valid")
 
         self.layer1 = self._make_layer(block, 64, layers[0])