赞
踩
目标检测算法根据其处理流程可以分为两大类:One-Stage(单阶段)算法和Two-Stage(两阶段)算法。以下是一些经典的目标检测算法:
单阶段算法:
两阶段算法:
这些算法各有优势,选择哪种算法通常取决于具体应用场景中对速度和精度的要求。
- from ultralytics import YOLO
-
- # 加载模型
- model = YOLO('E:\PycharmProjects\\yolov8n.pt')
-
- # 识别图片的物体
- results = model(['E:\\testImage\\cat1.png', 'E:\\testImage\\heixiong.jpg'])
-
- # 处理图片列表
- for result in results:
- boxes = result.boxes # Boxes object for bounding box outputs
- masks = result.masks # Masks object for segmentation masks outputs
- keypoints = result.keypoints # Keypoints object for pose outputs
- probs = result.probs # Probs object for classification outputs
- result.show() # display to screen
- result.save(filename='result.jpg') # save to disk
输出的结果图片:
模型下载地址:
安装:
pip install ultralytics
但是总是很糟心的timeout
换国内的源速度杠杠的,分分钟搞定!!!
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
国内的其他pip源:
# 清华
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
# 阿里
pip config set global.index-url https://mirrors.aliyun.com/pypi/simple/
# 腾讯
pip config set global.index-url http://mirrors.cloud.tencent.com/pypi/simple
# 豆瓣
pip config set global.index-url http://pypi.douban.com/simple/
如果PIP不可用?请检查PIP的路径是否加入到path(本说明仅限windows平台)
报错:
pip : 无法将“pip”项识别为 cmdlet
配置成功后测试:
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。