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

相关推荐
zm-v-159304339861 小时前
ArcGIS 水文分析升级:基于深度学习的流域洪水演进过程模拟
人工智能·深度学习·arcgis
拓端研究室2 小时前
视频讲解|核密度估计朴素贝叶斯:业务数据分类—从理论到实践
人工智能·分类·数据挖掘
灵智工坊LingzhiAI2 小时前
人体坐姿检测系统项目教程(YOLO11+PyTorch+可视化)
人工智能·pytorch·python
昨日之日20062 小时前
Video Background Remover V3版 - AI视频一键抠像/视频换背景 支持50系显卡 一键整合包下载
人工智能·音视频
SHIPKING3933 小时前
【机器学习&深度学习】什么是下游任务模型?
人工智能·深度学习·机器学习
子燕若水7 小时前
Unreal Engine 5中的AI知识
人工智能
极限实验室8 小时前
Coco AI 实战(一):Coco Server Linux 平台部署
人工智能
杨过过儿8 小时前
【学习笔记】4.1 什么是 LLM
人工智能
巴伦是只猫8 小时前
【机器学习笔记Ⅰ】13 正则化代价函数
人工智能·笔记·机器学习
大千AI助手8 小时前
DTW模版匹配:弹性对齐的时间序列相似度度量算法
人工智能·算法·机器学习·数据挖掘·模版匹配·dtw模版匹配