anirudh2290 closed pull request #12942: Fix the operator API documentation
URL: https://github.com/apache/incubator-mxnet/pull/12942
This is a PR merged from a forked repository.
As GitHub hides the original diff on merge, it is displayed below for
the sake of provenance:
As this is a foreign pull request (from a fork), the diff is supplied
below (as it won't show otherwise due to GitHub magic):
diff --git a/src/operator/contrib/multibox_target-inl.h
b/src/operator/contrib/multibox_target-inl.h
index f5a3b1189d1..daf870a1517 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;
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
With regards,
Apache Git Services
