MXNet中目标检测API使用总结：MultiBoxDetection、MultiBoxPrior、MultiBoxTarget

最新推荐文章于 2024-09-12 22:27:23 发布

XiangJiaoJun_

最新推荐文章于 2024-09-12 22:27:23 发布

阅读量3.1k

点赞数 4

分类专栏：深度学习文章标签： MXNet 物体检测

本文链接：https://blog.csdn.net/XiangJiaoJun_/article/details/84679596

版权

本文总结了MXNet中用于目标检测的三个关键API：MultiBoxPrior用于生成锚框，MultiBoxTarget进行锚框标记，MultiBoxDetection则处理预测锚框，包括非极大值抑制等步骤。通过对这些函数的深入理解和应用，可以简化目标检测任务的编程过程。

摘要由CSDN通过智能技术生成

MXNet在目标检测提供了许多API供用户调用，灵活使用这些函数能大大降低编程难度，其中跟锚框有关的三个函数MultiBoxDetection、MultiBoxPrior、MultiBoxTarget，我也探索了一段时间，下面总结一下这三个函数的用法吧，也方便自己查阅

MultiBoxPrior

该函数是用来在一张给定的特征图中，对于给定的大小比例和宽高比例，生成不同锚框，函数原型为mxnet.ndarray.contrib.MultiBoxPrior(data=None, sizes=_Null, ratios=_Null, clip=_Null, steps=_Null, offsets=_Null, out=None, name=None, **kwargs)下面总结一下一些常用参数的作用。

参数	作用说明
data	输入的特征图，一般形状为（批量大小，通道数、宽、高）
sizes	要生成的锚框的大小比例列表，比如 sizes=[0.2,0.5,0.75]
ratios	要生成的锚框的宽高比例列表，比如 ratios=[0.5,1,2]
clip	boolean 类型变量，设置是否要剪切超出边界的锚框，默认为0