p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!9452 fix a type error of yolov3-Darknet53 in pynative mode 

From: @lvchangquan
Reviewed-by: @chujinjin,@kisnwang
Signed-off-by: @chujinjin,@kisnwang
This commit is contained in:
mindspore-ci-bot 2020-12-03 21:34:56 +08:00 committed by Gitee
parent b5269d6bd4 d0c01f3856
commit dfaae73afd
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
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)