今天,人工智能正影响我们生产、生活的方方面面。10月10日,为期三天的2018华为全联接大会在上海拉开帷幕,此次大会以“+智能,见未来”为主题,发布了AI战略及全球领先的全栈全场景AI解决方案,并推出全球首个覆盖全场景人工智能的昇腾系列IP和芯片,打造全面领先的AI实力。而我们作为Python程序员在人工智能领域的技术与贡献也是举重若轻的,下面来了解一下怎么训练自己的目标检测模型。

目标检测是AI的一项重要应用,通过目标检测模型能在图像中把人、动物、汽车、飞机等目标物体检测出来,甚至还能将物体的轮廓描绘出来,就像下面这张图,是不是很酷炫呢,嘿嘿
在动手训练自己的目标检测模型之前,建议先了解一下目标检测模型的原理,这样才会更加清楚模型的训练过程。
本文将在我们搭建好的AI实战基础环境上,基于SSD算法,介绍如何使用自己的数据训练目标检测模型。SSD,全称Single Shot MultiBox Detector(单镜头多盒检测器),是Wei Liu在ECCV 2016上提出的一种目标检测算法,是目前流行的主要检测框架之一。
本案例要做的识别便是在图像中识别出熊猫,可爱吧,呵呵
下面按照以下过程介绍如何使用自己的数据训练目标检测模型:
1、安装标注工具
要使用自己的数据来训练模型,首先得先作数据标注,也就是先要告诉机器图像里面有什么物体、物体在位置在哪里,有了这些信息后才能来训练模型。
(1)标注数据文件
目前流行的数据标注文件格式主要有VOC_2007、VOC_2012,该文本格式来源于Pascal VOC标准数据集,这是衡量图像分类识别能力的重要基准之一。本文采用VOC_2007数据格式文件,以xml格式存储,如下:

其中重要的信息有:
filename:图片的文件名
name:标注的物体名称
xmin、ymin、xmax、ymax:物体位置的左上角、右下角坐标
(2)安装标注工具
如果要标注的图像有很多,那就需要一张一张手动去计算位置信息,制作xml文件,这样的效率就太低了。
所幸,有一位大神开源了一个数据标注工具labelImg,可以

本文介绍了如何使用Python和深度学习训练目标检测模型,特别是SSD算法。首先讲解了目标检测的重要性,然后详细阐述了从数据标注、数据集划分、SSD模型配置、预训练模型下载到模型训练的全过程,最后展示了模型的使用方法。
最低0.47元/天 解锁文章

3813

被折叠的 条评论
为什么被折叠?



