柬埔寨东方明珠EasyDL16687259995
afksadklfhk 发布于2020-04-09 11:19 浏览:134 回复:1
1
收藏

EasyDL专业版支持飞桨Master模式,内置百度海量数据训练的预训练模型,可在少量训练数据上达到更优的训练效果。目前上线了FasterR-CNN,YOLOv3,SSD三种检测模型,我们对比三种模型在coco2017权威目标检测数据集上的性能,从表1可以看出FasterR-CNN和YOLOv3在模型精度上比较有优势,但SSD在模型推理速度上非常有优势。用户可以根据不同的应用场景去选择不同的模型。近期专业版会陆续上线更多性能更加优秀的开源模型和百度自研大规模预训练模型。
1
表1.coco2017模型性能对比[1]
我们分别在三个检测模型上训练下货架数据,可以看出使用FasterR-CNN训练的效果会比其他两个模型好一些。从精度提升角度,我们选用FasterR-CNN。
2
表2.货架挡板检测模型型优化效果[1]
2)代码级灵活调参
EasyDL专业版提供给用户灵活的代码级调参能力,方便有一定深度学习基础的用户对模型进行更细致灵活地定制。
以FasterR-CNN模型为例,用户不仅可以对预设定的系统参数(如模型的训练迭代数(epochs)、模型输入尺寸(inputsize)、学习率衰减策略(lrdecay))等进行调节,也可以针对模型的配置代码及参数进行自由定制。
限于篇幅,我们简单对其中的一些参数进行介绍。
模型输入尺寸(inputsize):模型的输入尺寸指的是训练数据实际输入到模型中的尺寸,一般来说大的输入尺寸有助于模型对小目标的学习,但大的输入尺寸会降低模型的推理速度,因此该参数的调节要根据实际去选择。
学习率衰减策略((lrdecay):学习率是模型训练中非常重要的超参数,训练初期配置较大学习率,训练后期逐步降低学习率可以帮助模型收敛到更优解。用户可通过配置lr_decay=True来使用学习率衰减策略。
Anchor配置参数(aspect_ratios):基于anchor原理的物体检测模型对于先验的anchor高宽比的选择有一定要求,基于数据集特性去调整anchor参数可以让模型更好地学习到定位物体的能力。
针对货架挡板高宽比较极端的特性,我们对代码模板的anchor参数进行调整。调整策略可以选用对标注数据聚类等方法,我们通过自研的高宽比自适应能力对宽高比进行优化,发现特定的aspect_ratios修改如下图3可以很好的适应货架挡板数据集,再次训练模型,如图4,FasterR-CNN模型效果从0.812提升到0.87。后续高宽比自适应的能力会以可选组件的方式开放给用户。

收藏
点赞
1
个赞
共1条回复 最后由付洋洋carrie回复于2020-04-09 16:26
#2付洋洋carrie回复于2020-04-09 16:26:31

手动赞

1
TOP
切换版块