【承接图像分类、检测、分割、生成相关项目,私信。】
基于MXNet gluon 的SSD模型训练
官方教程链接:SSD 使用Gluon
整理版的SSD代码博文:MXNet SSD检测模型
1. 数据准备
车辆检测数据集(友情提示1积分,如无积分。可私信)
请下载上述数据集,内部包含图片+工具代码+lst文件例子。
##2. 数据集制作
官方文档中是这样写的:
但是,在实际操作过程中我发现,A需要设置为4才好用。具体,可以参考 数据集里的文件。
3.模型训练
3.1 训练代码
通过修改data_shape 和数据集