未验证 提交 54b39949 编写于 作者: Q qingqing01 提交者: GitHub

Merge pull request #5919 from qingqing01/roi_pooling

Fix bug in RoI pooling.
...@@ -13,6 +13,7 @@ See the License for the specific language governing permissions and ...@@ -13,6 +13,7 @@ See the License for the specific language governing permissions and
limitations under the License. */ limitations under the License. */
#include "ROIPoolLayer.h" #include "ROIPoolLayer.h"
#include <cfloat>
namespace paddle { namespace paddle {
...@@ -126,10 +127,8 @@ void ROIPoolLayer::forward(PassType passType) { ...@@ -126,10 +127,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.
先完成此消息的编辑!
想要评论请 注册