计算机系统应用教程网站

网站首页 > 技术文章 正文

物体分割检测YOLO4算法环境配置 yolo实例分割

btikc 2024-10-19 03:19:14 技术文章 12 ℃ 0 评论

安装Opencv

opencv网址:https://opencv.org/releases/

环境变量配置

系统变量:OpenCV_DIR    D:\BlackCatSoftWare\opencv\opencv

再进入Path,添加:

D:\BlackCatSoftWare\opencv\opencv\build\x64\vc16\bin
D:\BlackCatSoftWare\opencv\opencv\build\x64\vc16\lib

安装 CUDA

cuda下载:https://developer.nvidia.com/cuda-toolkit-archive

查看适合的cuda版本:

我的系统支持10.2版本。

自定义,全部勾选:

检测安装成功:nvcc -V

安装cuDNN

cuDNN网址:https://developer.nvidia.com/rdp/cudnn-archive

需要注册账号登录后才可以下载。

将解压后的三个文件夹复制到Cuda对应目录下。

安装CMAKE

cmake网址:https://cmake.org/download/

安装成功:

编译darknet

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


Path中添加环境变量:C:\Users\Cat\Desktop\darknet\build\darknet\x64

下载权重文件yolov4.weights放在darknet目录下

修改文件,将cuda改为当前版本号:

<ImportGroup Label="ExtensionSettings">
    <Import Project="$(VCTargetsPath)\BuildCustomizations\CUDA 10.2.props" />
</ImportGroup>

<ImportGroup Label="ExtensionTargets">
    <Import Project="$(VCTargetsPath)\BuildCustomizations\CUDA 10.2.targets" />
</ImportGroup>

VS2022打开项目:

1.将上方解决方案配置和解决方案平台更改为Release和x64

2.引入opencv

windows上进行训练:darknet.exe detector train data/obj.data data/yolo-obj.cfg data/yolov4.conv.137 -map

Tags:

本文暂时没有评论,来添加一个吧(●'◡'●)

欢迎 发表评论:

最近发表
标签列表