This is an automated email from the ASF dual-hosted git repository.
anirudh2290 pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/master by this push:
new af55104 Fix the operator API documentation (#12942)
af55104 is described below
commit af551045a4a26d8d12d7c2b23edc2cba3c8163e9
Author: Lin Yuan <[email protected]>
AuthorDate: Wed Oct 24 11:05:44 2018 -0700
Fix the operator API documentation (#12942)
* Fix the operator API documentation
* update message
---
src/operator/contrib/multibox_target-inl.h | 14 +++++++-------
1 file changed, 7 insertions(+), 7 deletions(-)
diff --git a/src/operator/contrib/multibox_target-inl.h
b/src/operator/contrib/multibox_target-inl.h
index f5a3b11..daf870a 100644
--- a/src/operator/contrib/multibox_target-inl.h
+++ b/src/operator/contrib/multibox_target-inl.h
@@ -218,15 +218,15 @@ class MultiBoxTargetProp : public OperatorProperty {
CHECK_EQ(in_shape->size(), 3) << "Input: [anchor, label, clsPred]";
TShape ashape = in_shape->at(mboxtarget_enum::kAnchor);
CHECK_EQ(ashape.ndim(), 3) << "Anchor should be batch shared N*4 tensor";
- CHECK_EQ(ashape[0], 1) << "Anchors are shared across batches, first dim=1";
- CHECK_GT(ashape[1], 0) << "Number boxes should > 0";
- CHECK_EQ(ashape[2], 4) << "Box dimension should be 4:
[xmin-ymin-xmax-ymax]";
+ CHECK_EQ(ashape[0], 1) << "Anchors are shared across batches, first
dimension should be 1";
+ CHECK_GT(ashape[1], 0) << "Number boxes should be greater than 0";
+ CHECK_EQ(ashape[2], 4) << "Box dimension should be 4: [xmin, ymin, xmax,
ymax]";
TShape lshape = in_shape->at(mboxtarget_enum::kLabel);
- CHECK_EQ(lshape.ndim(), 3) << "Label should be [batch-num_labels-(>=5)]
tensor";
- CHECK_GT(lshape[1], 0) << "Padded label should > 0";
- CHECK_GE(lshape[2], 5) << "Label width must >=5";
+ CHECK_EQ(lshape.ndim(), 3) << "Label should be [batch, num_labels,
label_width] tensor";
+ CHECK_GT(lshape[1], 0) << "Padded label should be greater than 0";
+ CHECK_GE(lshape[2], 5) << "Label width should be greater than or equal to
5";
TShape pshape = in_shape->at(mboxtarget_enum::kClsPred);
- CHECK_EQ(pshape.ndim(), 3) << "Prediction:
[nbatch-num_classes-num_anchors]";
+ CHECK_EQ(pshape.ndim(), 3) << "Prediction: [batch, num_classes,
num_anchors]";
CHECK_EQ(pshape[2], ashape[1]) << "Number of anchors mismatch";
TShape loc_shape = Shape2(lshape[0], ashape.Size()); // batch - (num_box
* 4)
TShape lm_shape = loc_shape;