惯性聚合 高效追踪和阅读你感兴趣的博客、新闻、科技资讯
阅读原文 在惯性聚合中打开

推荐订阅源

罗磊的独立博客
SecWiki News
SecWiki News
酷 壳 – CoolShell
酷 壳 – CoolShell
爱范儿
爱范儿
量子位
M
MIT News - Artificial intelligence
GbyAI
GbyAI
cs.AI updates on arXiv.org
cs.AI updates on arXiv.org
K
KPMG report finds enterprise disconnect between AI and its ROI | CIO
TaoSecurity Blog
TaoSecurity Blog
博客园 - 【当耐特】
H
Heimdal Security Blog
腾讯CDC
The Last Watchdog
The Last Watchdog
Security Archives - TechRepublic
Security Archives - TechRepublic
Hacker News: Ask HN
Hacker News: Ask HN
S
Schneier on Security
Microsoft Security Blog
Microsoft Security Blog
WordPress大学
WordPress大学
博客园 - 司徒正美
Recent Commits to openclaw:main
Recent Commits to openclaw:main
C
Cybersecurity and Infrastructure Security Agency CISA
S
SegmentFault 最新的问题
大猫的无限游戏
大猫的无限游戏
Application and Cybersecurity Blog
Application and Cybersecurity Blog
F
Full Disclosure
有赞技术团队
有赞技术团队
T
Tailwind CSS Blog
Engineering at Meta
Engineering at Meta
cs.CV updates on arXiv.org
cs.CV updates on arXiv.org
T
Threatpost
月光博客
月光博客
A
Arctic Wolf
OSCHINA 社区最新新闻
OSCHINA 社区最新新闻
雷峰网
雷峰网
T
Troy Hunt's Blog
freeCodeCamp Programming Tutorials: Python, JavaScript, Git & More
The Cloudflare Blog
D
DataBreaches.Net
O
OpenAI News
L
LINUX DO - 最新话题
宝玉的分享
宝玉的分享
小众软件
小众软件
V
Vulnerabilities – Threatpost
A
About on SuperTechFans
人人都是产品经理
人人都是产品经理
T
The Exploit Database - CXSecurity.com
Martin Fowler
Martin Fowler
美团技术团队
P
Privacy International News Feed

博客园 - 你看起来真的很好吃

linux系统时间管理 安装历史docker版本 OCR文本提取+NER命名实体识别 测试 深度学习-激活函数 go——make和new的区别 go——GC垃圾回收机制 go——标识符的命名规范 go语言——数据类型 go语言——转义符 go开发环境安装 git使用 项目文档目录总结 鼠标连点器——python版 高匿名动态IP代理获取 logging模块进行格式化输出 笔记 Django admin 添加操作记录 facenet + fiass 实现人脸识别
windows使用YOLOV5训练模型——搭建编译环境
你看起来真的很好吃 · 2023-12-13 · via 博客园 - 你看起来真的很好吃

1:安装Nvidia显卡驱动、cuda和cuDNN

1.1下载安装Navida显卡驱动


NAVIDIA驱动下载地址:https://www.nvidia.cn/Download/index.aspx

 下载完成后,在CMD中输入  nvidia-smi 验证是否安装成功

如果有错误

2:下载CUDA(本人下载12.2版本)

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

3:下载CUDNN

cudnn下载地址(注意cuda版本):https://developer.nvidia.cn/rdp/cudnn-archive

 4:安装CUDA

4.1运行cuda,建议使用默认路径

4.2:安装完成后设置环境变量

 计算机属性  -> 系统 -> 系统信息 -> 高级系统设置 -> 系统 的环境变量中可以看到多了CUDA_PATH和CUDA_PATH_V12_2两个环境变量

接下来在系统中添加已下几个环境变量

cuda默认安装路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2

CUDA_SDK_PATH:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2
CUDA_LIB_PATH:%CUDA_PATH%\lib\x64
CUDA_BIN_ATH:%CUDA_PATH%\bin
CUDA_SDK_BIN_PATH:%CUDA_SDK_PATH%\bin\win64(12.2版本不存在该文件)
CUDA_SDK_LIB_PATH:%CUDA_SDK_PATH%\common\lib\x64(12.2版本不存在该文件)

5:安装CUDNN

将cuCNN解压,将bin、include、lib文件复制到cuda中(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2)

6:cuda安装测试

打开cmd输入:>nvcc --version

即可看到cuda信息

7:安装anaconda 

anaconda下载地址:https://www.anaconda.com/download/

Anaconda3-2023.09-0-Windows-x86_64.exe 下载

链接:https://pan.baidu.com/s/1YfzwXMyJ_tT8ti0KcspzEg
提取码:8888

创建Python环境名YOLOv5  python版本3.10

conda添加豆瓣源:conda config --add channels http://pypi.douban.com/simple/

conda添加阿里源:conda config --add channels https://mirrors.aliyun.com/pypi/simple/

conda添加清华源: conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free

conda添加中科源: conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/

添加后使用命令: conda config --show channels  查看源是否添加成功

8:安装yolov5环境

进入环境yolov5,通过命令安装环境

pip install -r requirements.txt

9:下载预训练权重文件

yolov5-6.2 权重文件链接:https://pan.baidu.com/s/1D7Cf98WHEMQJFOjuQJL0lQ
提取码:8888

项目中新建weights文件夹用于存放权重文件

9:执行测试代码

权重文件下载地址: https://github.com/ultralytics/yolov5/releases/tag/v6.2

https://github.com/ultralytics/yolov5/releases/download/v6.2yolov5n.pt 
https://github.com/ultralytics/yolov5/releases/download/v6.2/yolov5s.pt 
https://github.com/ultralytics/yolov5/releases/download/v6.2lov5m.pt 
https://github.com/ultralytics/yolov5/releases/download/v6.2/yolov5l.pt 
https://github.com/ultralytics/yolov5/releases/download/v6.2/yolov5x.pt 

指定权重文件、设置置信度
python D:\projects\yolov5-6.2\detect.py --weights D:\projects\yolov5-6.2\weights\yolov5s.pt --conf 0.4

识别后的图片会存储在D:\projects\yolov5-6.2\runs\detect 文件夹下