目标检测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检测的图片

相关推荐
铮铭3 分钟前
【论文阅读】具身人工智能:从大型语言模型到世界模型
论文阅读·人工智能·语言模型
en-route11 分钟前
从零开始学神经网络——CNN(卷积神经网络)
人工智能·神经网络·cnn
Niuguangshuo14 分钟前
深度学习:池化(Pooling)
人工智能·深度学习
元基时代22 分钟前
专业的短视频发布矩阵哪家靠谱
大数据·人工智能·python·矩阵
腾讯数据架构师1 小时前
k8s 兼容摩尔线程
人工智能·云原生·容器·kubernetes·cube-studio
wuk9981 小时前
MATLAB的CFAR(恒虚警率)图像目标检测
目标检测·matlab·目标跟踪
春末的南方城市1 小时前
AI视频生成进入多镜头叙事时代!字节发布 Waver 1.:一句话生成 10 秒 1080p 多风格视频,创作轻松“一键”达!
人工智能·深度学习·机器学习·计算机视觉·aigc
机器之心2 小时前
节前重磅:开源旗舰模型新SOTA,智谱GLM-4.6问世
人工智能·openai
肖书婷2 小时前
人工智能-机器学习day2
人工智能·机器学习
西猫雷婶2 小时前
pytorch基本运算-torch.normal()函数生成的随机数据添加噪声
人工智能·pytorch·python·深度学习·学习·线性代数·机器学习