使用YOLOv3训练自己的数据集

1、下载源码

git clone https://github.com/pjreddie/darknet

或者直接到github上下载 

2、存放自己的数据集到对应的路径

 

  

3、修改dataset.yaml

 

4、修改 yolov3.yaml

 

5、下载requirement.txt种的包 

可以使用下面的语句在终端下载

pip install （包名） -i https://pypi.tuna.tsinghua.edu.cn/simple

6、修改train.py 

修改路径为下面内容

修改训练的相关参数（根据自己的情况而定） 

 

 

7、运行过程出现下面的报错内容解决办法

OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.
OMP: Hint This means that multiple copies of the OpenMP runtime have been linked into the program. That is dangerous, since it can degrade performance or cause inc
orrect results. The best thing to do is to ensure that only a single OpenMP runtime is linked into the process, e.g. by avoiding static linking of the OpenMP runti
me in any library. As an unsafe, unsupported, undocumented workaround you can set the environment variable KMP_DUPLICATE_LIB_OK=TRUE to allow the program to contin
ue to execute, but that may cause crashes or silently produce incorrect results. For more information, please see http://www.intel.com/software/products/support/.

在运行的文件里面添加以下内容

import os
os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"

 8、测试，想要输出图片（有锚框的图片结果）

修改detect.py

 

 

8、总结 

以上是使用yolov3训练自己数据集的过程，有什么问题欢迎在评论区告知我