您的位置:首页 > 大数据 > 人工智能

【AI实战】手把手教你训练自己的目标检测模型(SSD篇)

2020-07-20 14:55 375 查看

目标检测是AI的一项重要应用,通过目标检测模型能在图像中把人、动物、汽车、飞机等目标物体检测出来,甚至还能将物体的轮廓描绘出来,就像下面这张图,是不是很酷炫呢,嘿嘿


在动手训练自己的目标检测模型之前,建议先了解一下目标检测模型的原理(见文章:大话目标检测经典模型RCNN、Fast RCNN、Faster RCNN,以及Mark R-CNN),这样才会更加清楚模型的训练过程。

本文将在我们前面搭建好的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,可以通过可视化的操作界面进行画框标注,就能自动生成VOC格式的xml文件了。该工具是基于Python语言编写的,这样就支持在Windows、Linux的跨平台运行,实在是良心之作啊。安装方式如下:
a. 下载源代码
 

http://blog.tianya.cn/post-7807122-131380620-1.shtml
http://blog.tianya.cn/post-7807123-131380622-1.shtml
http://blog.tianya.cn/post-7807124-131380623-1.shtml
http://blog.tianya.cn/post-7807125-131380627-1.shtml
http://blog.tianya.cn/post-7807126-131380629-1.shtml
http://blog.tianya.cn/post-7807127-131380632-1.shtml
http://blog.tianya.cn/post-7807129-131380633-1.shtml
http://blog.tianya.cn/post-7807130-131380634-1.shtml
http://blog.tianya.cn/post-7807132-131380638-1.shtml
http://blog.tianya.cn/post-7807133-131380641-1.shtml
http://blog.tianya.cn/post-7807134-131380642-1.shtml
http://blog.tianya.cn/post-7807135-131380646-1.shtml
http://blog.tianya.cn/post-7807136-131380649-1.shtml
http://blog.tianya.cn/post-7807138-131380651-1.shtml
http://blog.tianya.cn/post-7807139-131380654-1.shtml
http://blog.tianya.cn/post-7807140-131380656-1.shtml
http://blog.tianya.cn/post-7807141-131380657-1.shtml
http://blog.tianya.cn/post-7807142-131380660-1.shtml
http://blog.tianya.cn/post-7807143-131380661-1.shtml
http://blog.tianya.cn/post-7807144-131380662-1.shtml
http://blog.tianya.cn/post-7807145-131380667-1.shtml
http://blog.tianya.cn/post-7807146-131380668-1.shtml
http://blog.tianya.cn/post-7807147-131380673-1.shtml
http://blog.tianya.cn/post-7807148-131380674-1.shtml
http://blog.tianya.cn/post-7807149-131380677-1.shtml
http://blog.tianya.cn/post-7807150-131380678-1.shtml
http://blog.tianya.cn/post-7807151-131380682-1.shtml
http://blog.tianya.cn/post-7807152-131380683-1.shtml
http://blog.tianya.cn/post-7807153-131380684-1.shtml
http://blog.tianya.cn/post-7807156-131380753-1.shtml
http://blog.tianya.cn/post-7807157-131380754-1.shtml
http://blog.tianya.cn/post-7807159-131380757-1.shtml
http://blog.tianya.cn/post-7807158-131380760-1.shtml
 

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐