IntenseWebs/openvino
3
0 Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Revert PSROIPooling average logic introduced in 305f0056059b091e0290b… (#3765)

Browse Source 
* Revert PSROIPooling average logic introduced in 305f005605

* fix computing bin coords in average mode
This commit is contained in:
Mateusz Tabaka authored and GitHub committed 2021-01-11 17:30:53 +03:00
1 parent 0d22328a25
commit 1e418ca549
3 files changed
+31 -24
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

inference-engine/src/mkldnn_plugin/nodes/psroi.cpp
+4 -4
View File
@@ -102,10 +102,10 @@ public:
roi_width = roi_end_w - roi_start_w;
roi_height = roi_end_h - roi_start_h;
} else if (mode_ == "average") {
roi_start_w = round(bottom_rois[1] * spatial_scale_);
roi_start_h = round(bottom_rois[2] * spatial_scale_);
roi_end_w = round(bottom_rois[3] * spatial_scale_) + 1.0f;
roi_end_h = round(bottom_rois[4] * spatial_scale_) + 1.0f;
roi_start_w = static_cast<float>(round(bottom_rois[1])) * spatial_scale_;
roi_start_h = static_cast<float>(round(bottom_rois[2])) * spatial_scale_;
roi_end_w = static_cast<float>(round(bottom_rois[3]) + 1.0f) * spatial_scale_;
roi_end_h = static_cast<float>(round(bottom_rois[4]) + 1.0f) * spatial_scale_;
// Force too small ROIs to be 1x1
roi_width = std::max<float>(roi_end_w - roi_start_w, 0.1f); // avoid 0
roi_height = std::max<float>(roi_end_h - roi_start_h, 0.1f);