从今年四月份开始报名参加了软件杯,参赛题目是基于人工智能和机器视觉的安全帽提取和分析。因为这个题目有幸接触到了yolov2–实时目标检测框架。
关于yolov2的介绍各位只要去上网搜一下就可以查到很多相关资料介绍。今天详细介绍一下如何用yolov2训练自己的数据。
官方提供的windows下的训练教程
以下是训练详细步骤
-
下载yolov2代码
yolov2 代码网址 -
配置yolov2环境:vs2015+opencv3.4+cuda8.0+cudnn7
(1)步骤1完成之后在darknet目录下找到darknet.sln,用VS2015打开之后darknet.sln后导入opencv(导入opencv教程)
(2)接下来导入cuda8.0.由于官方建议的是cuda9.1版本,如果不是使用该版本应该先打开darknet.vcxproj修改第55行和第288行将CUDA 9.1.targets改为CUDA 8.0.targets,保存。(vs2015配置cuda教程)
ÿ