深度学习实战 | WIN10环境下配置Darknet并使用Yolov4-tiny训练Gun-and-Sword数据集

目录

一、准备工作

(一)环境配置

1. cuda和cudnn

2. Visual Studio配置

3. opencv安装

(二)Darknet安装

1. 下载源文件

2. 编译项目

二、使用Yolov4-tiny进行训练

(一)数据集标注

(二)模型配置

1. 模型文件修改

2. 下载预训练权重

(三)模型训练

(四)模型测试及效果

三、参考资料


一、准备工作

(一)环境配置

1. cuda和cudnn

配置cuda11.1+cudnn v8.2.1,这里提供相应的下载链接,可以去官网下最新的,安装步骤都差不多。

cuda11.1:链接:https://pan.baidu.com/s/17FsL7v255mRw_IGMXflaFQ  提取码:4ycb 

cudnn v8.2.1:链接:https://pan.baidu.com/s/1LNtWPkXZk4dRbMRhFPGUaQ  提取码:6drh 

下载好之后先点开cuda_11.1.0_456.43_win10.exe,出现下图

 选定一个目录作为cuda的临时存放目录,点击"OK",等待提取完毕。

 等待提取完毕后会弹出安装界面

 点击“同意并继续”,再选择自定义,简单粗暴,全部勾选上

 下一步选择安装位置,默认在C盘。记住这个位置,点击下一步进行安装并等待安装完毕。

 安装完cuda后找到刚刚安装cuda的位置,我这里是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1,将cudnn-11.3-windows-x64-v8.2.0.53.zip中cuda文件夹中的bin,include,lib目录下的文件复制到上面那个目录中对应的bin,include,lib中即可完成配置。在cmd中输入nvcc -V可查看cuda版本。

2. Visual Studio配置

Visual Studio主要用来编译Darknet生成可执行的Darknet.exe,本文使用VS2015。

VS2015下载:Visual Studio Older Downloads - 2017, 2015 & Previous Versions (microsoft.com)

找到下面找到VS2015然后点“下载”

 找到Visual Studio Community 2015再点“Download”就可以下载VS2015的镜像了。

 下好之后用UltraISO把镜像里的文件提取出来,装好UltraISO之后右键镜像就会有UltraISO的选项,点击它就可以提取文件了。提取出来之后,文件如下图。

 点击“vs_community”,选好路径完成安装即可,要是提示有些包安装不了可以在安装界面那里设置搜索包的路径为packages目录。

3. opencv安装

下载链接:Download OpenCV from SourceForge.net

下载完后打开选择文件提取路径即可,后面编译darknet时会用到。我这里把它提取到了C:\Users\jiebi\Downloads\opencv。

(二)Darknet安装

1. 下载源文件

本文使用AlexeyAB版的Darknet。

AB版darknet下载链接:GitHub - AlexeyAB/darknet: YOLOv4 / Scaled-YOLOv4 / YOLO - Neural Networks for Object Detection (Windows and Linux version of Darknet )

下载后把它解压到桌面,文件大致内容如下。

2. 编译项目

 进入bulid下的darknet文件夹中,用编辑器编辑darknet.vcxproj,找到下图的两个地方,将CUDA后的数字改为11.1(如果不是11.1的话)。

 然后进入C:\Program Files (x86)\MSBuild\Microsoft.Cpp\v4.0\BuildCustomizations,查看是否有下图四个文件,没有的话就到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\extras\visual_studio_integration\MSBuildExtensions下拷贝过来。

接着配置opencv。用vs2015打开darknet.sln,右键右边的darknet,点击“属性”。

 在顶部“配置”和“平台”选择release和x64,再选中左边的“VC++”目录。选中右边的“包含目录”,点击右边的下拉箭头,选择“编辑”。

 在上面的框输入前面opencv提取目录下的bulid\include,bulid\include\opencv和bulid\include\opencv2,如下图,然后点击确定。

继续在darknet属性页下选中“库目录”,  在上面的框输入前面opencv提取目录下的bulid\x64\vc14\lib。

 点击确定后回到属性页,在右边找到“链接器”,点开并选中“输入”,选中右边的“附加依赖项”,点击右边的下拉箭头进入编辑。在上面的框框中输入opencv提取目录中\build\x64\vc14\lib下的库文件(没有d结尾的那个)。

 点击确定,回到IDE,点击选项卡中的“生成”--“生成darknet”,等待编译完成。完成后会在darknet-master\build\darknet目录下生成x64文件夹,里面就有我们需要用到的darknet.exe。

二、使用Yolov4-tiny进行训练

(一)数据集标注

本文使用VOTT来标注数据集。

VOTT下载链接:https://github.com/microsoft/VoTT/releases/download/v1.7.0/vott-win.exe

下载完后打开会自动安装。

进入主界面,在File--Open Image Directory中选择图片所在目录,然后选择好Region Type并输入图片物体标签后点击continue。

 用鼠标在图片上选取目标所在的区域,然后在左下角选择目标对应的标签。点击双箭头可切换到下一张图片。全部标记完后点击Object Detection -- Export Tag,选择输出格式为YOLO,然后点击“Export”导出。

VOTT生成的文件如下

进入data目录

最后将data目录下的文件都拷贝到x64文件夹下的data目录中。

(二)模型配置

(这里我配置了好几个模型训练的效果都不太好,所以参考了https://blog.csdn.net/weixin_43850253/article/details/118220395的模型)

1. 模型文件修改

将x64文件夹下的yolov4-tiny复制一份,重命名为yolov4-tiny-mytrain.cfg。修改文件中subdivisions的值为8,max_batches为6000,steps为4800,5000,yolo层前一个卷积层的filters为21(Ctrl+F搜索yolo)以及yolo层的classes为2,保存。文件内容如下。

[net]
# Testing
#batch=1
#subdivisions=1
# Training
batch=64
subdivisions=8 
width=416
height=416
channels=3
momentum=0.9
decay=0.0005
angle=0
saturation = 1.5
exposure = 1.5
hue=.1

learning_rate=0.00261
burn_in=1000

max_batches = 6000
policy=steps
steps=4800,5400
scales=.1,.1

[convolutional]
batch_normalize=1
filters=32
size=3
stride=2
pad=1
activation=leaky

[convolutional]
batch_normalize=1
filters=64
size=3
stride=2
pad=1
activation=leaky

[convolutional]
batch_normalize=1
filters=64
size=3
stride=1
pad=1
activation=leaky

[route]
layers=-1
groups=2
group_id=1

[convolutional]
batch_normalize=1
filters=32
size=3
stride=1
pad=1
activation=leaky

[convolutional]
batch_normalize=1
filters=32
size=3
stride=1
pad=1
activation=leaky

[route]
layers = -1,-2

[convolutional]
batch_normalize=1
filters=64
size=1
stride=1
pad=1
activation=leaky

[route]
layers = -6,-1

[maxpool]
size=2
stride=2

[convolutional]
batch_normalize=1
filters=128
size=3
stride=1
pad=1
activation=leaky

[route]
layers=-1
groups=2
group_id=1

[convolutional]
batch_normalize=1
filters=64
size=3
stride=1
pad=1
activation=leaky

[convolutional]
batch_normalize=1
filters=64
size=3
stride=1
pad=1
activation=leaky

[route]
layers = -1,-2

[convolutional]
batch_normalize=1
filters=128
size=1
stride=1
pad=1
activation=leaky

[route]
layers = -6,-1

[maxpool]
size=2
stride=2

[convolutional]
batch_normalize=1
filters=256
size=3
stride=1
pad=1
activation=leaky

[route]
layers=-1
groups=2
group_id=1

[convolutional]
batch_normalize=1
filters=128
size=3
stride=1
pad=1
activation=leaky

[convolutional]
batch_normalize=1
filters=128
size=3
stride=1
pad=1
activation=leaky

[route]
layers = -1,-2

[convolutional]
batch_normalize=1
filters=256
size=1
stride=1
pad=1
activation=leaky

[route]
layers = -6,-1

[maxpool]
size=2
stride=2

[convolutional]
batch_normalize=1
filters=512
size=3
stride=1
pad=1
activation=leaky

##################################

[convolutional]
batch_normalize=1
filters=256
size=1
stride=1
pad=1
activation=leaky

[convolutional]
batch_normalize=1
filters=512
size=3
stride=1
pad=1
activation=leaky

[convolutional]
size=1
stride=1
pad=1
filters=21
activation=linear

[yolo]
mask = 3,4,5
anchors = 10,14,  23,27,  37,58,  81,82,  135,169,  344,319
classes=2
num=6
jitter=.3
scale_x_y = 1.05
cls_normalizer=1.0
iou_normalizer=0.07
iou_loss=ciou
ignore_thresh = .7
truth_thresh = 1
random=0
resize=1.5
nms_kind=greedynms
beta_nms=0.6

[route]
layers = -4

[convolutional]
batch_normalize=1
filters=128
size=1
stride=1
pad=1
activation=leaky

[upsample]
stride=2

[route]
layers = -1, 23

[convolutional]
batch_normalize=1
filters=256
size=3
stride=1
pad=1
activation=leaky

[convolutional]
size=1
stride=1
pad=1
filters=21
activation=linear

[yolo]
mask = 1,2,3
anchors = 10,14,  23,27,  37,58,  81,82,  135,169,  344,319	
classes=2
num=6
jitter=.3
scale_x_y = 1.05
cls_normalizer=1.0
iou_normalizer=0.07
iou_loss=ciou
ignore_thresh = .7
truth_thresh = 1
random=0
resize=1.5
nms_kind=greedynms
beta_nms=0.6

2. 下载预训练权重

下载https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v4_pre/yolov4-tiny.conv.29到x64文件夹中。

(三)模型训练

在shell中输入

darknet.exe detector train data/obj.data cfg/yolov4-tiny-mytrain.cfg yolov4-tiny.conv.29 -map

开始训练。训练时会同时绘制损失曲线和map变化曲线。我训练的模型效果如下。

(四)模型测试及效果

将yolov4-tiny-mytrain.cfg复制一份然后改名为yolov4-tiny-mytest.cfg,并且将yolov4-tiny-mytest.cfg中的batch和subdivisions都改为1。

在x64文件夹下打开shell,执行

darknet.exe detector map data/obj.data cfg/yolov4-tiny-mytest.cfg backup/yolov4-tiny-mytrain_best.weights > map.txt

此时x64目录下会生成map.txt文件,里面的内容大致如下。

 CUDNN_HALF=1 
net.optimized_memory = 0 
mini_batch = 1, batch = 1, time_steps = 1, train = 0 
Create CUDA-stream - 0 
 Create cudnn-handle 0 
nms_kind: greedynms (1), beta = 0.600000 
nms_kind: greedynms (1), beta = 0.600000 

 seen 64, trained: 352 K-images (5 Kilo-batches_64) 

 calculation mAP (mean average precision)...
 Detection layer: 30 - type = 28 
 Detection layer: 37 - type = 28 

 detections_count = 29, unique_truth_count = 12  
 rank = 0 of ranks = 29 
class_id = 0, name = gun, ap = 80.56%   	 (TP = 3, FP = 2) 
class_id = 1, name = sword, ap = 97.62%   	 (TP = 5, FP = 0) 

 for conf_thresh = 0.25, precision = 0.80, recall = 0.67, F1-score = 0.73 
 for conf_thresh = 0.25, TP = 8, FP = 2, FN = 4, average IoU = 56.40 % 

 IoU threshold = 50 %, used Area-Under-Curve for each unique Recall 
 mean average precision (mAP@0.50) = 0.890873, or 89.09 % 

Set -points flag:
 `-points 101` for MS COCO 
 `-points 11` for PascalVOC 2007 (uncomment `difficult` in voc.data) 
 `-points 0` (AUC) for ImageNet, PascalVOC 2010-2012, your custom dataset

 于是我们获得了模型关于gun的Average Precision为80.56%,关于sword的Average Precision为97.62%。

在x64的data目录下新建一个test文件夹,把新的数据拷到test文件夹里,然后在x64目录下打开shell,输入

darknet.exe detector test data/obj.data cfg/yolov4-tiny-mytest.cfg backup/yolov4-tiny-mytrain_best.weights

 会出现下面的界面

 输入测试图片的路径就能生成检测结果。

三、参考资料

手把手教你用AlexeyAB版Darknet - 知乎 (zhihu.com)

windows下darknet之yolo(gpu版本)安装 - 知乎 (zhihu.com)

有道云笔记 (youdao.com)

https://blog.csdn.net/weixin_43850253/article/details/118220395

### 回答1: 1. 首先,需要准备好训练数据集和标注文件,可以使用标注工具如LabelImg等进行标注。 2. 接着,需要下载darknet框架和yolov4-tiny模型的权重文件。 3. 在darknet目录下,修改配置文件yolov4-tiny.cfg,将其中的batch、subdivisions、classes、filters等参数根据自己的数据集进行修改。 4. 将训练数据集和标注文件放入darknet/data目录下。 5. 在命令行中输入以下命令进行训练: ./darknet detector train data/obj.data cfg/yolov4-tiny.cfg yolov4-tiny.weights 6. 训练过程中可以通过命令行输出的信息来观察训练情况,也可以使用工具如TensorBoard等进行可视化。 7. 训练完成后,可以使用训练好的模型进行目标检测,具体方法可以参考darknet官方文档。 ### 回答2: Darknet是一种流行的开源深度学习框架,它支持各种计算机视觉任务,例如对象检测、分类、语义分割等。Yolov4-tiny是一种基于深度神经网络的对象检测模型,其速度和准确性优于之前的版本,由于其轻量级的特性,适合在边缘设备上进行部署。在使用Darknet训练Yolov4-tiny时,需要以下几个步骤。 1. 数据集准备和标注 准备和标注数据集深度学习的第一步。数据集包括许多图像,每张图像上都标有框框来标注出对象的位置,同时还需要给每个对象打上标签。这通常需要使用专业的工具,例如LabelImg或VIA。数据集的质量和数量对于模型的准确性至关重要。 2. 修改配置文件 在训练模型之前,需要使用配置文件指定许多参数,例如学习率、训练迭代次数、批量大小等。这些参数的设置将直接影响模型的性能和训练时间。在Yolov4-tiny配置文件中,将输入图像的大小设置为416x416,并且根据自己的数据集修改标签类别数量、训练、验证和测试集路径等参数。此外,还可以尝试调整不同的超参数来改进模型性能。 3. 下载预训练权重 通常情况下,可以使用训练的权重来加速模型的训练。在Yolov4-tiny的情况下,可以从官方网站下载预训练的权重,并将其作为初始权重进行训练。 4. 开始训练模型 完成配置文件和权重下载后,可以使用Darknet开始训练模型。在命令行中输入相应的命令,包括配置文件路径、权重路径、数据集路径等。训练过程可能需要几个小时到几天,具体时间取决于数据集的大小和复杂性。 5. 评估训练模型 当模型训练完成后,可以使用测试集对模型进行评估,查看其在不同指标下的表现,例如mAP(mean average precision)。评估结果可以帮助了解模型训练的效果,以及在实际应用中模型的性能如何。 6. 部署模型 最后,可以将训练好的模型部署到边缘设备上进行使用,例如实时对象检测和跟踪。在部署模型时,需要考虑设备的性能和存储容量,并根据具体需求进行优化,例如芯片加速、量化等。 ### 回答3: YoloV4-tiny是一种物体检测算法模型,其可用于实现高效的实时目标检测应用。而darknet则是实现该模型训练深度学习框架,其可在Linux和Windows平台上运行。下面将详细介绍darknet训练YoloV4-tiny的步骤和方法。 首先,需要在计算机上安装darknet,可通过下载源代码后进行编译安装,也可直接使用已编译好的可执行文件。安装成功后,需要下载训练用的数据集,并将其转换为darknet可用的格式,通常为txt格式的标注文件和jpg格式的图片。将数据集放入darknet目录下的data文件夹中。 接下来,需要准备好yolov4-tiny配置文件。配置文件包括模型参数、训练参数、数据集路径等,可参照darknet自带的yolov4-tiny.cfg文件进行设置。其中,需要注意的是网络结构的参数需要与数据集的类别数目对应,否则会导致训练结果不准确。另外,还需设置学习率、批次大小、迭代次数等训练参数。 完成配置文件的设置后,就可开始训练模型了。在命令行中输入训练命令,如“./darknet detector train data/obj.data cfg/yolov4-tiny.cfg yolov4-tiny.weights -gpus 0,1”,其中参数含义分别为:数据集路径、配置文件路径、预训练权重路径、使用GPU设备和数量。 训练过程中,darknet会输出每一轮训练的损失值和预测精度,可通过观察损失值变化来判断训练进展情况。训练完成后,会在darknet目录下生成新的权重文件,可用于实际应用。 总的来说,训练YoloV4-tiny需要准备好数据集配置文件和训练参数,并在darknet中进行训练训练过程需要耗费一定时间和计算资源,但能得到高效、准确的检测模型,适用于各种物体检测应用。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值