Skip to content

Commit 4c9884e

Browse files
committed
refine unittest test=develop
1 parent 98c3294 commit 4c9884e

File tree

2 files changed

+15
-2
lines changed

2 files changed

+15
-2
lines changed

paddle/fluid/operators/roi_align_op.cc

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,20 @@ class ROIAlignOpMaker : public framework::OpProtoAndCheckerMaker {
132132
"and pooled_w, likewise for height")
133133
.SetDefault(-1);
134134
AddComment(R"DOC(
135-
135+
**RoIAlign Operator**
136+
137+
Region of interest align (also known as RoI align) is to perform
138+
bilinear interpolation on inputs of nonuniform sizes to obtain
139+
fixed-size feature maps (e.g. 7*7)
140+
141+
Dividing each region proposal into equal-sized sections with
142+
the pooled_width and pooled_height. Location remains the origin
143+
result.
144+
145+
In each ROI bin, the value of the four regularly sampled locations
146+
are computed directly through bilinear interpolation. The output is
147+
the mean of four locations.
148+
Thus avoid the misaligned problem.
136149
)DOC");
137150
}
138151
};

python/paddle/fluid/tests/unittests/test_roi_align_op.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -167,4 +167,4 @@ def test_check_output(self):
167167
self.check_output()
168168

169169
def test_check_grad(self):
170-
self.check_grad(['X'], 'Out', max_relative_error=0.005)
170+
self.check_grad(['X'], 'Out')

0 commit comments

Comments
 (0)