Browse Source

fix a bug of yolov3-Darknet53 in pynative mode

tags/v1.1.0
lvchangquan 5 years ago
parent
commit
d0c01f3856
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      model_zoo/official/cv/yolov3_darknet53/src/yolo.py

+ 2
- 2
model_zoo/official/cv/yolov3_darknet53/src/yolo.py View File

@@ -146,12 +146,12 @@ class YOLOv3(nn.Cell):
con1, big_object_output = self.backblock0(feature_map3)
con1 = self.conv1(con1)
ups1 = P.ResizeNearestNeighbor((img_hight / 16, img_width / 16))(con1)
ups1 = P.ResizeNearestNeighbor((img_hight // 16, img_width // 16))(con1)
con1 = self.concat((ups1, feature_map2))
con2, medium_object_output = self.backblock1(con1)
con2 = self.conv2(con2)
ups2 = P.ResizeNearestNeighbor((img_hight / 8, img_width / 8))(con2)
ups2 = P.ResizeNearestNeighbor((img_hight // 8, img_width // 8))(con2)
con3 = self.concat((ups2, feature_map1))
_, small_object_output = self.backblock2(con3)


Loading…
Cancel
Save