Browse Source

adaptation to Pooling class operator interface changes

tags/v1.2.1
liuluobin 4 years ago
parent
commit
679e856b7c
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      examples/model_security/model_attacks/cv/faster_rcnn/src/FasterRcnn/fpn_neck.py
  2. +1
    -1
      examples/model_security/model_attacks/cv/faster_rcnn/src/FasterRcnn/resnet50.py

+ 1
- 1
examples/model_security/model_attacks/cv/faster_rcnn/src/FasterRcnn/fpn_neck.py View File

@@ -89,7 +89,7 @@ class FeatPyramidNeck(nn.Cell):
self.interpolate1 = P.ResizeNearestNeighbor((48, 80))
self.interpolate2 = P.ResizeNearestNeighbor((96, 160))
self.interpolate3 = P.ResizeNearestNeighbor((192, 320))
self.maxpool = P.MaxPool(ksize=1, strides=2, padding="same")
self.maxpool = P.MaxPool(kernel_size=1, strides=2, pad_mode="same")

def construct(self, inputs):
x = ()


+ 1
- 1
examples/model_security/model_attacks/cv/faster_rcnn/src/FasterRcnn/resnet50.py View File

@@ -89,7 +89,7 @@ class ResNetFea(nn.Cell):
self.conv1 = _conv(3, 64, kernel_size=7, stride=2, padding=3, pad_mode='pad')
self.bn1 = _BatchNorm2dInit(64, affine=bn_training, use_batch_statistics=bn_training)
self.relu = P.ReLU()
self.maxpool = P.MaxPool(ksize=3, strides=2, padding="SAME")
self.maxpool = P.MaxPool(kernel_size=3, strides=2, pad_mode="SAME")
self.weights_update = weights_update
if not self.weights_update:


Loading…
Cancel
Save