Skip to content

P
Paddle

Crayon鑫 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 cc9a761a 编写于 作者: D dangqingqing
浏览文件
操作

Fix bug in RoI pooling.

上级 1f6002ed
隐藏空白更改
内联 并排
Showing with 2 addition and 4 deletion
paddle/gserver/layers/ROIPoolLayer.cpp
浏览文件 @ cc9a761a
...@@ -126,10 +126,8 @@ void ROIPoolLayer::forward(PassType passType) { ...@@ -126,10 +126,8 @@ void ROIPoolLayer::forward(PassType passType) {
bool isEmpty = (hend <= hstart) || (wend <= wstart); bool isEmpty = (hend <= hstart) || (wend <= wstart);
size_t poolIndex = ph * pooledWidth_ + pw; size_t poolIndex = ph * pooledWidth_ + pw;
if (isEmpty) { outputData[poolIndex] = isEmpty ? 0 : -FLT_MAX;
outputData[poolIndex] = 0; argmaxData[poolIndex] = -1;
argmaxData[poolIndex] = -1;
}
for (size_t h = hstart; h < hend; ++h) { for (size_t h = hstart; h < hend; ++h) {
for (size_t w = wstart; w < wend; ++w) { for (size_t w = wstart; w < wend; ++w) {
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或