目标检测——yolov5-3.1的环境搭建和运行

第一步:安装anaconda环境,并且配置好cuda,安装需要的基本包

查看对应cuda版本,后续下载cudatoolkit需要对应版本

bash 复制代码
nvcc -V

第二步:创建虚拟环境,激活环境,安装所需的包

bash 复制代码
conda create -n yolov5 python=3.8 
conda activate yolov5
conda install pytorch torchvision cudatoolkit=11.8 -c pytorch   

第三步:下载github文件,安装对应环境所需的包

下载网站:https://github.com/ultralytics/yolov5/releases/tag/v3.1

此处注意一定要到下载的文件目录下去!

bash 复制代码
cd D:\pythonProject\yolov5-3.1    
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple  

第四步:执行对应的detect.py文件

此处需要下载yolov5s.pt文件,放到weight目录下

bash 复制代码
python detect.py --source ./inference/images --weights weights/yolov5s.pt --conf 0.4

第五步:结果展示

inference下的image和output会显示结果。

相关推荐
AKAMAI13 小时前
安全风暴的绝地反击 :从告警地狱到智能防护
运维·人工智能·云计算
岁月宁静13 小时前
深度定制:在 Vue 3.5 应用中集成流式 AI 写作助手的实践
前端·vue.js·人工智能
galaxylove13 小时前
Gartner发布数据安全态势管理市场指南:将功能扩展到AI的特定数据安全保护是DSPM发展方向
大数据·人工智能
格林威14 小时前
偏振相机在半导体制造的领域的应用
人工智能·深度学习·数码相机·计算机视觉·视觉检测·制造
晓枫-迷麟15 小时前
【文献阅读】当代MOF与机器学习
人工智能·机器学习
来酱何人15 小时前
实时NLP数据处理:流数据的清洗、特征提取与模型推理适配
人工智能·深度学习·分类·nlp·bert
sensen_kiss15 小时前
INT301 Bio-computation 生物计算(神经网络)Pt.3 梯度下降与Sigmoid激活函数
人工智能·神经网络·机器学习
Shilong Wang15 小时前
MLE, MAP, Full Bayes
人工智能·算法·机器学习
数据库知识分享者小北15 小时前
云栖重磅|瑶池数据库:从云原生数据底座向“AI就绪”的多模态数据底座演进
数据库·人工智能·云原生
lingling00915 小时前
机械臂动作捕捉系统选型指南:从需求到方案,NOKOV 度量光学动捕成优选
人工智能·算法