目标检测yolo算法

yolov5s:

从github官网下载yolov5的算法之后,配置好环境(pycharm安装包-CSDN博客),再下载权重文件,比如默认的yolov5s.pt

运行当前文件(detect.py),就能看到exp文件夹中的结果

也可以自定义参数运行:

train.py和detect.py文件同时使用(训练自己的数据集)

先运行train.py(上面一行命令) 将其中的data.yaml(可自定义名字)中训练、验证的路径改成自己的数据库文件夹,权重文件看要求可大可小;

训练完之后生成best.pt和last.pt文件,

运行下面一行,导入检测图像和刚刚训练好的best.pt文件即可

yolov11:

常见虚拟环境、激活虚拟环境、下载torch、下载ultralytics:

直接下载出错:由于清华镜像源 (https://pypi.tuna.tsinghua.edu.cn) 返回了 HTTP 403 Forbidden,可能是镜像源暂时不可用或访问受限;

解决方法:pip install ultralytics --index-url https://pypi.org/simple

配置完成,切换成yolov11环境:

测试:

测试代码:(提前下载号yolo11n.pt或者别的pt文件)

python 复制代码
from ultralytics import YOLO

# 自动下载模型并推理(首次运行会下载yolov11n.pt)
model = YOLO('yolo11n.pt')
results = model.predict(r'C:\Users\18211\Desktop\ultralytics-main11\ultralytics-main\ultralytics\assets\bus.jpg')
results[0].show()  # 显示检测结果

结果:输出bus检测的图片

相关推荐
极客学术工坊1 天前
2023年第二十届五一数学建模竞赛-A题 无人机定点投放问题-基于抛体运动的无人机定点投放问题研究
人工智能·机器学习·数学建模·启发式算法
Theodore_10221 天前
深度学习(9)导数与计算图
人工智能·深度学习·机器学习·矩阵·线性回归
PPIO派欧云1 天前
PPIO上新GPU实例模板,一键部署PaddleOCR-VL
人工智能
TGITCIC1 天前
金融RAG落地之痛:不在模型,而在数据结构
人工智能·ai大模型·ai agent·ai智能体·开源大模型·金融ai·金融rag
chenzhiyuan20181 天前
《十五五规划》下的AI边缘计算机遇:算力下沉与工业智能化
人工智能·边缘计算
whaosoft-1431 天前
51c深度学习~合集11
人工智能
Tiandaren1 天前
大模型应用03 || 函数调用 Function Calling || 概念、思想、流程
人工智能·算法·microsoft·数据分析
像风一样的男人@1 天前
python --两个文件夹文件名比对(yolo 图和label标注比对检查)
windows·python·yolo
领航猿1号1 天前
Pytorch 内存布局优化:Contiguous Memory
人工智能·pytorch·深度学习·机器学习
综合热讯1 天前
宠智灵宠物识别AI:从犬猫到鸟鱼的全生态智能识别
人工智能·宠物