未验证 提交 9876bb09 编写于 作者: Y Yang yaming 提交者: GitHub

Merge pull request #5514 from pkuyym/fix-5513

Bug fix for ScaleSubRegionLayer
...@@ -49,7 +49,7 @@ void ScaleSubRegionLayer::forward(PassType passType) { ...@@ -49,7 +49,7 @@ void ScaleSubRegionLayer::forward(PassType passType) {
shape_ = TensorShape({batchSize, channelsNum_, imgH_, imgW_}); shape_ = TensorShape({batchSize, channelsNum_, imgH_, imgW_});
resetOutput(batchSize, imgV->getWidth()); resetOutput(batchSize, imgV->getWidth());
auto out = getOutput(); auto& out = getOutput();
out.setFrameHeight(imgH_); out.setFrameHeight(imgH_);
out.setFrameWidth(imgW_); out.setFrameWidth(imgW_);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册