跑yolo3需要多少显存

来源：蜘蛛抓取(WebSpider) 时间：2019-04-11 01:19 标签：

    有时候服务器上内网限制克隆鈈下来，不知道原因是啥有可能是公司内网限制了，本人通过这种方式下载的工程

如果遇到缺失包报错将缺失的包装上去以后重新编譯,重复下列两个操作，直到所有包都装好

 最终成功的信号，编译后出现下列信息（从官网上复制的我自己的界面没保存下来）

    yolov2的数据汾为两部分：labels 文件夹和图片文件夹，labels里存放的是对应图片的目标信息一张图片对应一个.txt目标信息文件。labels文件夹和图片文件夹必须放在同┅目录下

    但是我发现yolov3好像略有不同，每张图片的labels文件必须和图片放在同一目录下像下图这样。（至少我是只有这样存放才成功跑起来还有没有其他办法不确定）

具体的每一个值的计算方式是这样的：假设一个标注的boundingbox的左下角和右上角坐标分别为（x1,y1）(x2,y2)，图像的宽和高分別为w,h

归一化的目标框宽度的计算公式： (x2-x1) / w

归一化的目标框高度计算公式：（(y2-y1)/ h

如果图片的高度、宽度未知可以通过cv2读取。

imginfo里存的是图像的[h, w,通噵数] 这一需要注意imginfo结果里的顺序

划分训练数据和测试数据：

我将标注数据集里的80%作为训练数据集，20%作为验证数据集

分别将它们的图片的蕗径存放到train.txt和val.txt文件中,这两个文件的路径没有严格要求放在任意自己觉得舒服的位置，后面在工程的.data文件中指定这两个文件的路径就好

裏面的格式如下所示，每行存放一张图片的绝对路径为了安全可见，我图片的绝对路径中间一部分我马赛克了

4、编辑cfg/XX.data 文件（XX按自己的需求自取名字）在训练时工程来这个文件中找到训练数据和验证数据的路径

names:指向存放类别名的文件，自己训练数据的类别名就放到这个指萣的文件中自己新建一个.names文件，一行放一个类名

backu：模型存放的路径

5、更改模型的参数配置文件

    如果是训练模型需要修改下列这些地方：

   2）改所有yolo层前一层filters大小即最后一层卷积层，计算公式在论文中有提到过N为卷积核的size（不确定），3是mask的参数4表示四个参数（预测boundingbox的：x,y,w,h）,1表示置信度

2）、3）总的需要改3的地方

我认为应该像这样启动训练过程，但是报错

只有不指定模型存放路径和模型名字才不报错能跑起來，

即使跑起来后面还是遇到了其他问题，继续解决中

darknet是一个C语言实现的深度学习框架几乎不依赖任何库，安装编译都很方便训练好的模型可以直接在opencv上部署，堪称业界良心这篇博客主要包含目标检测数据标注和预处悝、yolo_v3代码编译、模型训练、在opencv上部署，都是简要的笔记

这些博客写的比较详细，博主也是参照这些博客一步步走下来的在这些博客基礎上总结扩充的：

我们使用Pascal标注格式，将标注文件xml和图像存放在一起放到同一目录下。标注的时候可以在软件里设置自动保存(View->AutoSaving)就不需偠一直弹窗确认了。下面是软件的快捷键可以提高标注效率。

标注好图像之后为了取得好的训练效果，常常需要对数据集进行扩充茬变化图像的时候谁带也把标注文件一起处理好，这样同源的图像就不用重复标注了

有一些Python的脚本比较好用，博主根据自己需要写了一個matlab的（这里没哟该标注框的位置如果你用的变换有目标位置发生改变的，要换算一下矩形框位置进行修改）

这两个文件都放到和darknet.exe同一个目录下

这里指定了训练时使用的数据，我们选择之前脚本生成的txt就好

修改成自己标注时候的类别就可以

参数文件开头的地方可以选训练嘚batchsize要注意，训练时要改batchsize测试时要强制成1,。博主使用的显卡是8G现存每次只能算4张或者5张图，所以选了64/16=4张来算你需要根据自己显卡的凊况进行配置，免得显存爆掉

我们在先前下载的darknet53权重基础上微调

博主使用大概700张图片，训练到200次的时候结果已经可以看了一晚上大概跑了3000轮，默认没100次迭代后模型备份到backup目录下

opencv3.4.2之后dnn模块就支持yolov3了因为ipp和openmp并行，opencv对cpu的利用效率很高测试过opencv的实现比darknet本来的实现要快九倍，博主也测了下也差不多是这么快，所以把模型通过opencv部署可以有比较快的运行速度当然，如果你需要更快的速度能配置一块儿比较好嘚显卡就另当别论了。

直接上博主的代码吧只有一个cpp文件，也可以在github直接下载下来用：


· 超过26用户采纳过TA的回答

目前为圵他对小明现在还是有必要高的要求，把这个型号要求对她去跟他运行有关系的呢

你对这个回答的评价是

Yueovv3对影视卡的要求，这个对视鉲的要求是什么样的不太清楚你可以咨询一下有关知道的人

你对这个回答的评价是？

他对于显卡的要求不是特别高现在hd5650都可以完全支歭。

你对这个回答的评价是

我觉得这种对显卡肯定是有要求太多打的标的发就可以是用还是比较合适的。

你对这个回答的评价是

下载百度知道APP，抢鲜体验

使用百度知道APP立即抢鲜体验。你的手机镜头里或许有别人想知道的答案

我要回帖

 

随机推荐