MXNET_DL_API

9.4 锚框

# 生成锚框
# data	: 图像样本（批量大小，通道数，高，宽）
# sizes	: 【锚框大小】
# ratios: 【宽高比】
# return: 多个锚框（批量大小，锚框数量，锚框坐标），搭配reshape((h,w,n+m-1,4))即可通过像素指定锚框。
mxnet.contrib.ndarray.MultiBoxPrior(data, sizes, ratios)

# 显示锚框
# axes	: plt.imshow()返回的图像
# bboxes: 【锚框坐标】-注意非归一化坐标
# label	: 【锚框标签】
# color	: 【锚框颜色】
d2l.show_bboxes(axes, bboxes, labels=None, colors=None):

# 标注锚框类别和偏移量，生成掩码。
# anchor: 【锚框】
# truth	: 【真值框】
# cls	: （批量大小, 包括背景的类别个数, 锚框数）
# return: 【锚框的偏移值、maks、锚框的分类标签】
# 均要添加样本维
mxnet.contrib.ndarray.MultiBoxTarget(anchor, truth, cls)

# 预测边界框，非极大抑制
# cls_probs		: 【类别预测概率】
# offset_preds	: 【锚框的偏移值】
# anchor		: 【锚框】
# threshold		: 【交并比阈值】
mxnet.contrib.ndarray.MultiBoxDetection(cls_probs, offset_preds, anchor, threshold)